torch>=2.0
transformers>=4.40
trl>=1.0
peft>=0.10
datasets>=2.0
deepspeed>=0.12
accelerate>=0.30
scikit-learn>=1.0
numpy>=1.24
pyyaml>=6.0
python-dotenv>=1.0
tqdm>=4.60

[all]
neotune[logging,ray]

[logging]
mlflow>=2.0
wandb>=0.15

[ray]
ray[train]>=2.0
