numpy
torch<2.3,>=2.0
omegaconf
rich
boto3
google-cloud-storage
tokenizers
packaging
cached_path>=1.6.2
transformers

[all]
ai2-olmo[dev,train]

[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

[train]
wandb
beaker-gantry
click
torchmetrics
smashed[remote]>=0.21.1
safetensors
datasets
scikit-learn
msgspec>=0.14.0
