einops~=0.6.0
torch>=1.8.0
torchscale~=0.2.0

[t5]
sentencepiece
transformers<4.32,>=4.5.0

[test]
black
kaleido
mypy
plotly
pre-commit
pytest
pytest-cov
ruff
xformers==0.0.20
