torch
einops

[all]
triton>=2.0.0
transformers>=4.38.0

[dev]
triton>=2.0.0
transformers>=4.38.0
pytest>=6.0
pytest-benchmark
numpy

[flex]
transformers>=4.38.0

[test]
pytest>=6.0
pytest-benchmark
numpy

[triton]
triton>=2.0.0
