openai~=1.55.3
tiktoken~=0.7.0
ratemate~=0.1
humanize~=4.8.0
xxhash~=3.4
numpy~=1.24
packaging>=22.0

[accelerate]
accelerate~=0.23.0

[anthropic]
anthropic~=0.40.0

[dev]
black~=23.10.0
ruff~=0.1.1
pytest~=7.4.2
pytest-cov~=4.1.0
build~=1.0.3
twine~=5.1.1

[docs]
mkdocs>=1.4.0
mkdocstrings[python]>=0.24.0
mkdocs-material>=9.0.0
mike>=1.1.2

[flash-attn]
flash_attn~=2.3.2

[hf]
torch~=2.3.0
transformers~=4.46.3

[ort]
optimum[onnxruntime]~=1.13.2

[ort-gpu]
optimum[onnxruntime-gpu]~=1.13.2

[quant]
bitsandbytes~=0.41.1

[xformers]
xformers~=0.0.22
