numpy<3,>=1.26
torch<2.14,>=2.10

[all]
torchtitan<0.3,>=0.2.2
megatron-core<0.19,>=0.18.2
deepspeed<0.20,>=0.19.4
ninja

[deepspeed]
deepspeed<0.20,>=0.19.4
ninja

[dev]
pytest
pytest-timeout
pre-commit
ruff==0.16.0

[megatron]
megatron-core<0.19,>=0.18.2

[torchtitan]
torchtitan<0.3,>=0.2.2
