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

[all]
xformers
black
isort
pre-commit
pytest

[dev]
black
isort
pre-commit
pytest

[xformers]
xformers
