einops>=0.6.1
matplotlib>=3.7.5
nbformat>=4.2.0
numpy>=1.23.5
pandas>=1.5.3
plotly>=5.24.0
pydantic>=2.9.0
scipy>=1.10.1
torch>=2.0.1
tqdm>=4.65.0
cloudpickle>=3.0.0

[dev]
pytest==7.4.3
transformers
datasets
transformer_lens

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

[tpu]
torch-xla[tpu]==2.3.0
torch==2.3.0
