adam_atan2==0.0.3
argdantic==1.3.3
coolname==2.2.0
einops==0.8.1
flash_attn @ https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.3/flash_attn-2.8.3+cu12torch2.7cxx11abiTRUE-cp310-cp310-linux_x86_64.whl#sha256=ce91e246f21d61ad66b1a7555340dbaa28e4aa86edcf00c18f0837422939b529
huggingface-hub==0.34.4
hydra-core==1.3.2
numba==0.61.2
omegaconf==2.3.0
pydantic==2.11.7
pydantic_core==2.33.2
pydantic-settings==2.14.1
torch==2.7.0+cu126
torchaudio==2.7.0+cu126
torchvision==0.22.0+cu126
tqdm==4.67.1
triton==3.3.0
wandb==0.21.3