einops
scikit-learn==1.6.1
umap-learn
pycolormap-2d
tqdm
torch==2.5.1
pytorch3d==0.7.8
torchdata==0.10.1
torchvision==0.20.1
