torch
numpy<=2.3.5
pixeltable
nvidia-ml-py
scikit-learn
umap-learn
torchvision
coreax
tqdm

[dev]
pytest
pre-commit
ruff
mypy
types-pillow
types-tqdm

[gpu]
jax[cuda]
