datasets>=2.5.0
einops
numpy>=1.20.0
prettytable>=3.5.0
protobuf==3.20.*
pynvml
scikit-learn>=1.0.0
sentencepiece==0.1.97
simple-parsing>=0.0.21
torch>=1.11.0
tqdm>=4.0.0
transformers>=4.0.0
jinja2

[dev]
hypothesis
pre-commit
pytest
pyright
