# Separate Linux x86_64 / CPython 3.12 environment; see docs/setup-27b.md.
# Install torch from the CUDA 12.8 index first. Never install this over the 4B venv.
torch==2.8.0
transformers==5.15.1
accelerate==1.12.0
numpy==1.26.4
matplotlib==3.10.8
bitsandbytes==0.50.2
huggingface-hub==1.33.0
tokenizers==0.22.2
safetensors==0.8.0
triton==3.4.0
fla-core==0.5.2
einops==0.8.1
jinja2==3.1.6
# Official prebuilt release wheel: CUDA 12, Torch 2.8, C++11 ABI, CPython 3.12.
# The URL fragment verifies the release asset SHA256 without enabling global --require-hashes.
causal-conv1d @ https://github.com/Dao-AILab/causal-conv1d/releases/download/v1.7.0/causal_conv1d-1.7.0%2Bcu12torch2.8cxx11abiTRUE-cp312-cp312-linux_x86_64.whl#sha256=a1fcd5d022b67ddfc9e1c291f890004c1c081a03d66b532f6eebe31219168e0e
