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

[:python_version < "3.11"]
tomli

[sae]
sae-lens
