numpy
torch>=2.0
torchmetrics
tokenizers
click
rich
cached-path
beaker-gantry
omegaconf
wandb
logzio-python-handler
boto3
google-cloud-storage
packaging
datasets
scikit-learn
smashed[remote]>=0.21.1
msgspec>=0.14.0
cached_path
transformers

[dev]
ruff
mypy<1.4,>=1.0
black<24.0,>=23.1
isort<5.13,>=5.12
pytest
pytest-sphinx
twine>=1.11.0
setuptools
wheel
build
