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

[all]
whittle[dev]
whittle[distributed]

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

[distributed]
deepspeed
triton==3.0.0
