numpy<2.1.0,>=1.21.5
packaging<24.2,>=21.3.0
torch<2.4.1,>=2.4.0
triton>=2.1.0
stanford-stk==0.7.1

[all]
pytest<8,>=7.2.1
coverage[toml]==7.4.4
grouped_gemm==0.1.6
pre-commit<4,>=3.4.0
pytest-cov<5,>=4
pytest_codeblocks<0.17,>=0.16.1
absl-py

[dev]
absl-py
coverage[toml]==7.4.4
pytest_codeblocks<0.17,>=0.16.1
pytest-cov<5,>=4
pytest<8,>=7.2.1
pre-commit<4,>=3.4.0

[gg]
grouped_gemm==0.1.6

[testing]
mosaicml>=0.24.1
