torch>=1.9.0
torchvision>=0.10.0
numpy>=1.21.0
pandas>=1.3.0
huggingface-hub>=0.16.0
safetensors>=0.3.0
jaxtyping>=0.2.0
beartype>=0.15.0

[all]
srmbench[dev,docs]

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=22.0.0
isort>=5.10.0
flake8>=5.0.0
mypy>=1.0.0
pre-commit>=2.20.0

[docs]
sphinx>=5.0.0
sphinx-rtd-theme>=1.0.0
myst-parser>=0.18.0
