--extra-index-url https://download.pytorch.org/whl/cu128

torch==2.9.0+cu128
torchvision==0.24.0+cu128
lightning==2.5.0.post0
accelerate==0.34.2
datasets==3.3.2
einops==0.8.1
fsspec==2024.2.0
huggingface_hub>=0.30
hydra-core==1.3.2
lm_eval==0.4.8
mup==1.0.0
omegaconf==2.3.0
PyYAML>=6.0
requests>=2.31
scikit-learn==1.5.1
sentencepiece>=0.2.0
tiktoken>=0.9.0
timm==0.9.16
tokenizers>=0.21.0
transformers==4.49.0
torchmetrics==1.6.2
tqdm>=4.66