torch
torchvision
torchmetrics
lightning
hydra-core
omegaconf
loguru
tabulate
pandas
rich
requests-cache
prettytable
submitit
timm

[all]
stable-pretraining[cluster,datasets,extras,tracking,vision,visualization]

[cluster]
hydra-submitit-launcher

[datasets]
datasets
pyarrow==20.0.0
minari[hdf5]>=0.5.3

[dev]
pytest
coverage
pytest-cov
codecov
pre-commit
ruff

[doc]
sphinx
sphinx-gallery
sphinx-book-theme
sphinxcontrib-bibtex
matplotlib
memory_profiler
myst-parser

[extras]
scikit-learn>=1.7.0
richuru

[tracking]
wandb

[vision]
transformers
pillow>=11.3.0

[visualization]
matplotlib
