rich
torch>=2.0.0
torchvision>=0.15.0
mup
litgpt
GitPython
wandb
neural-pipeline-search@ git+https://github.com/automl/neps.git

[dev]
ruff
pre-commit
