torch>=2.0
transformer_lens>=1.0
einops
numpy

[dev]
pytest>=7.0
