torch==2.6
torchdata
transformers==4.51.3
triton>=3.0
datasets>=3.3.0
einops
ninja
wandb
tiktoken
tensorboard
python-dotenv

[dev]
pytest
