numpy
torch
torchvision
torchaudio
jaxtyping
matplotlib
Pillow
scikit-learn
einops
rerun-sdk
minio
torchmetrics
termcolor
timm

[all]
xformers
black
isort
pre-commit
pytest
requests

[dev]
black
isort
pre-commit
pytest
requests

[xformers]
xformers
