einops>=0.6.1
matplotlib>=3.7.5
numpy>=1.23.5
pandas>=1.5.3
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

[tpu]
torch-xla>=2.0.1
