torch>=1.9.0
transformers>=4.46.0
jinja2>=3.1.0
tqdm>=4.50.0
scikit-learn>=0.24.0
pandas>=1.2.0
numpy>=1.21.0
numba>=0.56.0
datasets>=2.0.0
sentence-transformers>=2.0.0
faiss-cpu>=1.7.0
uncensorbench>=0.2.0
pebble>=5.0.0
latex2sympy2_extended>=1.0.0
sae_lens>=0.1.0
trl>=0.7.0
peft>=0.7.0
safetensors>=0.4.0
huggingface_hub>=0.20.0
psycopg2-binary>=2.9.0
pynndescent>=0.5.0
hdbscan>=0.8.0
umap-learn>=0.5.0
pacmap>=0.7.0
optuna>=3.0.0
hyperopt

[cuda]
flash-attn>=2.5.0

[harness]
lm-eval==0.4.8

[reft]
pyreft>=0.1.0

[sparsify]
sparsify>=0.1.0
