torch>=2.2
transformers>=4.51
accelerate>=0.27
safetensors>=0.7.0
textual>=0.50
pyyaml>=6.0

[bnb]
bitsandbytes>=0.43

[cuda]
bitsandbytes>=0.43
flash-attn>=2.5

[dev]
pytest>=8.0
httpx>=0.27
ruff>=0.6.9
pyright>=1.1.400
fastapi>=0.110
uvicorn[standard]>=0.44.0

[flash]
flash-attn>=2.5

[gguf]
gguf>=0.10

[hf]
datasets>=2.0

[pandas]
pandas>=3.0.2

[research]
datasets>=2.0
pandas>=3.0.2

[sae]
sae-lens<7.0,>=6.0

[serve]
fastapi>=0.110
uvicorn[standard]>=0.44.0
