torch>=2.1.0

[all]
triton>=2.2.0
prometheus-client>=0.19.0
transformers>=4.35.0
accelerate>=0.24.0

[dev]
pytest>=7.4.0
pytest-cov>=4.1.0
hypothesis>=6.88.0
black>=23.10.0
isort>=5.12.0
flake8>=6.1.0
mypy>=1.6.0
pre-commit>=3.5.0
transformers>=4.35.0
accelerate>=0.24.0
wandb>=0.16.0

[flash]
flash-attn>=2.4.0

[hf]
transformers>=4.35.0
accelerate>=0.24.0

[prometheus]
prometheus-client>=0.19.0

[triton]
triton>=2.2.0
