torch>=2.0.0
numpy>=1.24.0
tokenizers>=0.15.0
pyyaml>=6.0
tqdm>=4.60.0

[all]
USN[cuda,dev,docs]

[cuda]
triton>=2.1.0

[dev]
pytest>=7.0
pytest-cov
mypy
ruff
pre-commit
hypothesis>=6.0

[docs]
sphinx
sphinx-rtd-theme
