torch>=1.9.0
transformers>=4.20.0
tqdm>=4.50.0
scikit-learn>=0.24.0
pandas>=1.2.0
numpy>=1.21.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

[cuda]
flash-attn>=2.5.0

[harness]
lm-eval==0.4.8

[sparsify]
sparsify>=0.1.0
