torch>=2.0
ai2-tango[beaker,fairscale,gs,torch,transformers,wandb]>=1.3
torchmetrics==0.11.1
more_itertools
spacy>=3.0.0
wget
datasets>=2.14.6
accelerate
bettermap
tiktoken
protobuf<=3.20
bert_score
sacrebleu>=1.5.0
scikit-learn>=0.24.1
iso639>=0.1.4
rouge-score>=0.0.4
mypy_extensions
pytest
jinja2
pyyaml>=5

[dev]
isort
black
mypy
twine>=1.11.0
setuptools
wheel
packaging
pytest>=6.2.5
pytest-sphinx
pytest-xdist
pytest-forked
flaky
Sphinx<6
furo
myst-parser
sphinx-copybutton
sphinx-autobuild
sphinx-autodoc-typehints
