--extra-index-url https://download.pytorch.org/whl/cu121
torch==2.3.1+cu121
torchvision
lightning==2.2.5
ninja
jsonargparse[signatures]==4.28.0
numpy
scipy
wandb
fpsample
trimesh
zarr
scikit-learn
matplotlib
seaborn 
h5py
# pytorch3d @ git+https://github.com/facebookresearch/pytorch3d.git@stable
