torch>=2.1.0
transformers>=4.36.0
accelerate>=0.25.0
datasets>=2.14.0
unsloth>=2024.1
trl>=0.7.0
peft>=0.7.0
bitsandbytes>=0.41.0
vllm>=0.2.0
ollama>=0.1.0
llama-cpp-python>=0.2.0
pymupdf>=1.23.0
pdfplumber>=0.10.0
autoawq>=0.2.0
auto-gptq>=0.5.0
onnx>=1.15.0
gguf>=0.1.0
lm-eval>=0.4.0
lighteval>=0.4.0
pynvml>=11.5.0
codecarbon>=2.0.0
psutil>=5.9.0
gradio>=4.0.0
pyyaml>=6.0
tqdm>=4.66.0
rich>=13.0.0
click>=8.0.0

[all]
saara-ai[dev,docs,edge]

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
black>=23.0.0
ruff>=0.1.0
mypy>=1.0.0

[docs]
sphinx>=7.0.0
sphinx-book-theme>=1.0.0
mkdocs>=1.5.0
mkdocs-material>=9.0.0

[edge]
tensorrt>=8.6.0
onnxruntime-gpu>=1.16.0
