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

[:python_version < "3.11"]
tomli<3,>=2

[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]
build==1.5.0
detect-secrets==1.5.0
mypy==2.3.0
packaging==25.0
pip-audit==2.10.1
pre-commit==4.6.2
pytest==9.1.1
pytest-timeout==2.4.0
ruff==0.16.3
twine==6.2.0

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

[torchtitan]
torchtitan<0.3,>=0.2.2
