torch>=2.0
transformer_lens>=1.0
einops
numpy

[dev]
pytest>=7.0
sae-lens>=3.0

[sae]
sae-lens>=3.0
