transformer_lens<3,>=2.15
sae_lens>=6
torch
transformers<5,>=4.40
ir_datasets
numpy<2,>=1.26
datasets<4
pyarrow<21
jaxtyping
pandas

[axiomatic]
python-terrier
ir_axioms

[dev]
pytest
flake8
black==26.10.1

[examples]
fire
tqdm
python-terrier
pyterrier-dr

[plotting]
seaborn
matplotlib
plotly
streamlit

[rag]
datasets<4
ir_datasets
