lightning<3.0,>=2.3
torch<3.0,>=2.6
torchaudio<3.0,>=2.6
gin-config<1.0,>=0.5
einops<1.0,>=0.8
transformers<5.0,>=4.50

[dev]
ruff~=0.14
pytest

[train]
bayesian_optimization~=1.5.1
matplotlib
scikit_learn~=1.5.2
scipy~=1.14.1
wandb
