numpy
torch>=2.6.0
cached-path>=1.7.2
filelock<4.0,>=3.4
requests
packaging
dataclass-extensions>=0.3.0
pyyaml
rich
safetensors
importlib_resources
bettermap
pandas

[all]
ai2-olmo-core[beaker,comet,dev,dion,eval,fa4,fla,torchao,transformers,wandb]

[beaker]
beaker-py<3.0,>=2.5.4
beaker-gantry<4.0,>=3.4.3
GitPython<4.0,>=3.0
google-cloud-compute

[comet]
comet_ml

[dev]
black<24.0,>=23.1
boto3
build
furo==2024.8.6
google-cloud-storage
isort<5.14,>=5.12
matplotlib
mypy<1.4,>=1.0
myst-parser<3.0,>=1.0
pytest
pytest-memray
pytest-sphinx
pytest-xdist
ruff==0.15.9
setuptools
Sphinx<=8.1.3,>=6.0
sphinx-autobuild
sphinx-autodoc-typehints==1.23.3
sphinx-copybutton
sphinx-inline-tabs
twine>=1.11.0
wheel

[dion]

[dion:sys_platform == "linux"]
dion

[eval]
ai2-olmo-eval==0.9.0

[fa4]

[fa4:sys_platform == "linux"]
flash-attn-4>=4.0.0b4

[fla]
flash-linear-attention==0.5.2

[torchao]
torchao==0.15.0

[transformers]
transformers>=5.4.0

[wandb]
wandb
