pyyaml<7,>=6.0
tqdm<5,>=4.62.3
torchmetrics<1.7.5,>=1.0
torch_optimizer<0.4,>=0.3.0
torchvision<0.22.1,>=0.21.0
torch<2.7.1,>=2.6.0
requests<3,>=2.26.0
numpy<2.3.0,>=1.21.5
psutil<8,>=5.8.0
coolname<3,>=1.1.0
tabulate==0.9.0
py-cpuinfo<10,>=8.0.0
packaging<25.1,>=21.3.0
importlib-metadata<9,>=5.0.0
mosaicml-cli<0.8,>=0.5.25
pillow<12,>=10.3.0

[all]
ipython==8.37.0
py-cpuinfo<10,>=8.0.0
mosaicml-streaming<1.0
sphinxcontrib-images==0.9.4
sphinx-argparse==0.4.0
sphinx_panels==0.6.0
nbsphinx==0.9.1
paramiko<4,>=3.4.0
databricks-sdk<1,>=0.50.0
sphinx_markdown_tables==0.0.17
sphinx-copybutton==0.5.2
sphinxcontrib-htmlhelp==2.0.0
pytest_codeblocks==0.17.0
sphinxcontrib-devhelp==1.0.0
setuptools<79.0.0
slack_sdk<4,>=3.19.5
sphinxcontrib.katex==0.9.11
transformers!=4.34.0,<4.54,>=4.11
sphinxcontrib-applehelp==1.0.0
pre-commit<5,>=3.4.0
sentencepiece==0.2.0
sphinx==4.4.0
tensorboard<3.0.0,>=2.9.1
fasteners==0.18
onnx<2,>=1.12.0
mock-ssh-server==0.9.1
custom_inherit==2.4.1
neptune<2.0.0,>=1.6.2
ipykernel==6.29.5
sphinxext.opengraph==0.9.1
cryptography==44.0.3
jupyter==1.1.1
yamllint==1.37.1
testbook==0.4.2
sphinxcontrib-qthelp==1.0.0
sphinxcontrib-serializinghtml==1.1.5
pypandoc==1.15
moto[s3]<6,>=5.0.1
boto3<2,>=1.21.45
datasets<5,>=2.4
oci<3.0.0,>=2.88.2
mlflow<3.0,>=2.14.1
coverage[toml]<8,>=7.8.0
protobuf<5.30
GitPython==3.1.44
pycocotools<3,>=2.0.4
recommonmark==0.7.1
pytest-httpserver<1.2,>=1.0.4
myst-parser==0.16.1
sphinxemoji==0.2.0
wandb<0.19,>=0.13.2
huggingface-hub<0.34,>=0.21.2
google-cloud-storage<4.0,>=2.0.0
pytest==7.4.4
pandas<3.0,>=2.0.0
pynvml<12,>=11.5.0
apache-libcloud<4,>=3.3.1
peft<0.17,>=0.10.0
junitparser==4.0.2
pandoc==2.4
traitlets==5.14.3
onnxruntime<2,>=1.12.1
docutils==0.17.1
furo==2022.9.29
comet_ml<4.0.0,>=3.31.12

[base]

[coco]
pycocotools<3,>=2.0.4

[comet_ml]
comet_ml<4.0.0,>=3.31.12

[databricks]
databricks-sdk<1,>=0.50.0

[dev]
custom_inherit==2.4.1
junitparser==4.0.2
coverage[toml]<8,>=7.8.0
fasteners==0.18
pytest==7.4.4
ipython==8.37.0
ipykernel==6.29.5
jupyter==1.1.1
yamllint==1.37.1
recommonmark==0.7.1
sphinx==4.4.0
pre-commit<5,>=3.4.0
docutils==0.17.1
sphinx_markdown_tables==0.0.17
sphinx-argparse==0.4.0
sphinxcontrib.katex==0.9.11
sphinxcontrib-applehelp==1.0.0
sphinxcontrib-devhelp==1.0.0
sphinxcontrib-htmlhelp==2.0.0
sphinxcontrib-serializinghtml==1.1.5
sphinxcontrib-qthelp==1.0.0
sphinxext.opengraph==0.9.1
sphinxemoji==0.2.0
furo==2022.9.29
sphinx-copybutton==0.5.2
testbook==0.4.2
myst-parser==0.16.1
sphinx_panels==0.6.0
sphinxcontrib-images==0.9.4
pytest_codeblocks==0.17.0
traitlets==5.14.3
nbsphinx==0.9.1
pandoc==2.4
pypandoc==1.15
GitPython==3.1.44
moto[s3]<6,>=5.0.1
mock-ssh-server==0.9.1
cryptography==44.0.3
pytest-httpserver<1.2,>=1.0.4
setuptools<79.0.0

[gcs]
google-cloud-storage<4.0,>=2.0.0

[libcloud]
apache-libcloud<4,>=3.3.1

[mlflow]
mlflow<3.0,>=2.14.1
databricks-sdk<1,>=0.50.0
pynvml<12,>=11.5.0

[mlperf]
py-cpuinfo<10,>=8.0.0

[neptune]
neptune<2.0.0,>=1.6.2

[nlp]
transformers!=4.34.0,<4.54,>=4.11
datasets<5,>=2.4
huggingface-hub<0.34,>=0.21.2

[oci]
oci<3.0.0,>=2.88.2

[onnx]
onnx<2,>=1.12.0
onnxruntime<2,>=1.12.1

[pandas]
pandas<3.0,>=2.0.0

[peft]
peft<0.17,>=0.10.0

[sentencepiece]
protobuf<5.30
sentencepiece==0.2.0

[slack]
slack_sdk<4,>=3.19.5

[streaming]
mosaicml-streaming<1.0
boto3<2,>=1.21.45
paramiko<4,>=3.4.0

[system_metrics_monitor]
pynvml<12,>=11.5.0

[tensorboard]
tensorboard<3.0.0,>=2.9.1

[wandb]
wandb<0.19,>=0.13.2
