transformers>=4.37.0
torch
triton
einops

[dev]
transformers>=4.37.0
torch
triton
einops
black==21.4b0
flake8>=3.8.3
isort>=5.5.4
pre-commit
pytest
pytest-xdist

[quality]
black==21.4b0
flake8>=3.8.3
isort>=5.5.4
pre-commit
pytest
pytest-xdist
