torch # --index-url https://download.pytorch.org/whl/cu118
torchvision # --index-url https://download.pytorch.org/whl/cu118
lightning
torchmetrics
hydra-core
hydra-colorlog
hydra-optuna-sweeper
rootutils
dacite
rich
opencv-python
wandb
transformers
numpy<2.0.0 # note that numpy should be less than 2
plyfile
einops
jaxtyping
kornia
git+https://github.com/Lightning-AI/torchmetrics.git
git+https://github.com/rmurai0610/diff-gaussian-rasterization-w-pose.git
git+https://github.com/nerfstudio-project/gsplat.git