torch>=2.0.0
numpy>=1.21.0

[all]
dsalt[triton]
dsalt[flash-attn]
dsalt[dev]
dsalt[docs]

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=22.0.0
isort>=5.10.0
flake8>=4.0.0
mypy>=1.0.0
pre-commit>=2.20.0

[docs]
sphinx>=5.0.0
sphinx-rtd-theme>=1.2.0
myst-parser>=0.18.0

[flash-attn]
flash-attn>=2.0.0

[triton]
triton>=2.0.0
