torch>=2.2
transformers>=4.51
accelerate>=0.27
safetensors>=0.7.0
textual>=0.50
pyyaml>=6.0
fastapi>=0.110
uvicorn[standard]>=0.44.0

[cuda]
bitsandbytes>=0.43
kernels>=0.12

[cuda-experimental]
bitsandbytes>=0.43
kernels>=0.12
flash-attn>=2.5

[dev]
pytest>=8.0
httpx>=0.27
ruff>=0.6.9
pyright>=1.1.400

[flash]
flash-attn>=2.5

[gguf]
gguf>=0.10

[hf]
datasets>=2.0

[notebook]
plotly>=5.0
pandas>=3.0.2
kaleido>=0.2

[pandas]
pandas>=3.0.2

[research]
datasets>=2.0
pandas>=3.0.2

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