jsonschema
numpy
psutil
scikit-learn
sentencepiece
torch
transformers>=4.40

[sae]
sae-lens
