torch>=2.0
torchvision>=0.17.0
torchaudio>=2.2.0
datasets>=2.16.0
transformers>=4.40.0
sentencepiece>=0.1.99
tokenizers>=0.15.2
accelerate>=0.27.0
tqdm>=4.66.0
numpy>=1.24.0
pandas>=2.0.0
pyyaml>=6.0
scikit-learn>=1.3.0
einops>=0.7.0
wandb>=0.16.0

[dev]
pytest>=7.0.0
