torch>=1.8.0
einops~=0.6.0
torchscale~=0.2.0
xformers<0.0.22,>=0.0.20

[all]
black
flake8
isort
kaleido
mypy
plotly
pytest
pytest-cov
pre-commit

[dev]
pre-commit
black
flake8
isort
kaleido
mypy
plotly
pytest
pytest-cov

[test]
black
flake8
isort
kaleido
mypy
plotly
pytest
pytest-cov
