torch>=2.0.0
numpy>=1.20.0
einops>=0.7.0

[all]
triton>=2.0.0
transformers>=4.30.0
datasets>=2.0.0
pytest>=7.0.0
black>=23.0.0
ruff>=0.1.0

[dev]
pytest>=7.0.0
black>=23.0.0
ruff>=0.1.0
ipython>=8.0.0

[gpu]
triton>=2.0.0
