wheel
torch==2.9.1
diffusers==0.33.0
nn-dataset>=2.2.8
nn-rag>=2.2.4
nn-dup>=2.2.4
bitsandbytes>=0.45.0
peft>=0.14.0
gguf>=0.10.0
trl
sentence-transformers>=3.3.1
faiss-cpu
overrides>=7.7.0
datasketch
langgraph>=1.0.4
unsloth==2026.3.4
unsloth-zoo==2026.3.2
triton>=3.4.0
matplotlib
optimum[onnxruntime-gpu]
onnx>=1.16.0
onnxruntime-gpu>=1.18.0
kernels

[flash]
flash_attn

[stat]
nn-stat
