pandas
matplotlib
typing-extensions
torch>=2.5
transformers<4.54,>=4.51.3
litgpt[extra]==0.5.12
lm-eval<0.4.9.1
syne-tune[moo]==0.13.0
torchvision>=0.18
boto3==1.34.147
botocore==1.34.147
wandb

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

[profiling]
nvidia-ml-py
psutil
