pandas
matplotlib
typing-extensions
torch>=2
transformers>=4
litgpt[all]==0.5.0
syne-tune[moo]>=0.13
torchvision>=0.18
boto3==1.34.147
botocore==1.34.147
deepspeed

[dev]
build
pre-commit
pytest>=8
ruff
mypy
commitizen
mike
mkdocs
mkdocs-material
mkdocs-autorefs
mkdocs-gen-files
mkdocs-literate-nav
mkdocstrings[python]
black
