# Core
torch
torchvision
torchmetrics
numpy<2
einops
tqdm
omegaconf
hydra-core

# Data loading / IO
h5py
tensordict
opencv-python
pillow
imageio
pandas
albumentations

# Geometry / metrics
scikit-learn
scikit-image
lpips

# Logging / visualization
matplotlib
plotly
tabulate
tensorboard
wandb
viser
nerfview

# CUDA extensions -- pip install from source with --no-build-isolation:
#   gsplat:     git+https://github.com/JoannaCCJH/gsplat.git
#   fused-ssim: git+https://github.com/rahul-goel/fused-ssim
