einops>=0.6
numpy>=1.23
pycolormap-2d>=1.0
pytorch3d>=0.7
scikit-learn>=1.2
torch>=2.0
tqdm>=4.0
umap-learn>=0.5

[dev]
matplotlib>=3.7
pytest>=7.0
ruff>=0.4
