torch>=2.1
numpy>=1.24
matplotlib>=3.5

[data]
datasets>=2.18
pyarrow>=14
zstandard>=0.22
tokenizers>=0.15

[dev]
auralite-ai[data,full,lint,rag,serve,studio,test]

[export]
onnx>=1.15
onnxruntime>=1.17
onnxscript>=0.5

[full]
transformers>=4.46
peft>=0.13
safetensors>=0.4
accelerate>=1.1
sentencepiece>=0.2
protobuf>=4.21
tiktoken>=0.7
pydantic>=2

[full:platform_system != "Darwin"]
bitsandbytes>=0.43

[gguf]
llama-cpp-python>=0.3

[hf]
transformers>=4.46
peft>=0.13
accelerate>=1.1

[lint]
ruff>=0.5
pyright>=1.1

[qlora]
auralite-ai[hf]

[qlora:platform_system != "Darwin"]
bitsandbytes>=0.43

[rag]
sentence-transformers>=3
faiss-cpu>=1.8

[safetensors]
safetensors>=0.4

[serve]
fastapi>=0.111
uvicorn[standard]>=0.30

[studio]
PySide6-Essentials>=6.6

[test]
pytest>=7
pytest-cov>=4
hypothesis>=6
pytest-benchmark>=4
fastapi>=0.111
uvicorn>=0.30
httpx>=0.27
pydantic>=2
httpx2>=2
pyyaml>=6
setuptools>=77
openai>=1.40

[torchao]
torchao>=0.10

[vllm]

[vllm:platform_system == "Linux"]
vllm>=0.5
