einops>=0.6.1
matplotlib>=3.7.5
numpy>=1.23.5
pandas>=1.5.3
torch>=2.0.1
tqdm>=4.65.0
cloudpickle>=3.0.0

[dev]
pytest==7.4.3
transformers
datasets
transformer_lens
syrupy

[docs]
sphinx
sphinx-math-dollar
sphinx-rtd-theme
sphinx-autobuild
myst_parser

[vis]
plotly>=5.20.0
