torch>=2.3.0
datasets==3.6.0
einops==0.8.1
entmax==1.3
fsspec==2025.3.0
huggingface_hub==0.36.0
hf_transfer==0.1.9
hydra-core==1.3.2
lightning==2.6.0
numpy==2.3.2
omegaconf==2.3.0
packaging==25.0
Requests==2.32.4
rich==14.2.0
scipy==1.16.3
timm==1.0.22
tokenizers==0.22.1
torchmetrics==1.8.2
tqdm==4.67.1
transformers==4.57.3
triton==3.4.0
wandb==0.23.1
nvitop==1.3.2
# flash-attn must be installed separately due to build flags:
# pip install flash-attn==2.8.3 --no-build-isolation