pyyaml<7,>=6.0
tqdm<5,>=4.62.3
torchmetrics<1.1,>=0.10.0
torch_optimizer<0.4,>=0.3.0
torchvision<0.17,>=0.13.1
torch<2.1.1,>=1.13.1
requests<3,>=2.26.0
numpy<1.27.0,>=1.21.5
psutil<6,>=5.8.0
coolname<3,>=1.1.0
tabulate==0.9.0
py-cpuinfo<10,>=8.0.0
packaging<23,>=21.3.0
importlib-metadata<7,>=5.0.0
mosaicml-cli<0.6,>=0.5.8

[all]
mock-ssh-server==0.9.1
pycocotools<3,>=2.0.4
moto[s3]<5,>=4.0.1
cryptography==41.0.4
fasteners==0.18
pynvml<12,>=11.5.0
pypandoc==1.11
sphinx-copybutton==0.5.2
sphinxemoji==0.2.0
custom_inherit==2.4.1
comet_ml<4.0.0,>=3.31.12
py-cpuinfo<10,>=8.0.0
coverage[toml]==7.3.0
docutils==0.17.1
protobuf<3.21
junitparser==3.1.0
sphinxext.opengraph==0.8.2
jupyter==1.0.0
ipykernel==6.25.2
google-cloud-storage<3.0,>=2.0.0
transformers<4.34,>=4.11
sphinx_panels==0.6.0
sphinxcontrib.katex==0.9.6
wandb<0.16,>=0.13.2
sphinx_markdown_tables==0.0.17
testbook==0.4.2
setuptools<=59.5.0
sphinx-argparse==0.4.0
sentencepiece==0.1.99
datasets<3,>=2.4
deepspeed==0.8.3
oci<3.0.0,>=2.88.2
slack_sdk<4,>=3.19.5
mlflow<3.0,>=2.5.0
vit_pytorch==0.35.8
toml==0.10.2
scikit-learn<2,>=1.0.1
recommonmark==0.7.1
pytest_codeblocks==0.16.1
nbsphinx==0.9.1
ipython==8.11.0
GitPython==3.1.37
timm<0.6,>=0.5.4
pydantic<2,>=1.0
pytest==7.4.2
furo==2022.9.29
tensorboard<3.0.0,>=2.9.1
pandas<3.0,>=2.0.0
pre-commit<4,>=3.4.0
sphinxcontrib-images==0.9.4
onnxruntime<2,>=1.12.1
pytest-httpserver<1.1,>=1.0.4
paramiko<3,>=2.11.0
boto3<2,>=1.21.45
sphinx==4.4.0
onnx<2,>=1.12.0
yamllint==1.32.0
pandoc==2.3
mosaicml-streaming<1.0
myst-parser==0.16.1
apache-libcloud<4,>=3.3.1
monai<1.3,>=0.9.1
traitlets==5.10.0

[base]

[coco]
pycocotools<3,>=2.0.4

[comet_ml]
comet_ml<4.0.0,>=3.31.12

[deepspeed]
deepspeed==0.8.3
pydantic<2,>=1.0

[dev]
custom_inherit==2.4.1
junitparser==3.1.0
coverage[toml]==7.3.0
fasteners==0.18
pytest==7.4.2
toml==0.10.2
ipython==8.11.0
ipykernel==6.25.2
jupyter==1.0.0
yamllint==1.32.0
recommonmark==0.7.1
sphinx==4.4.0
pre-commit<4,>=3.4.0
docutils==0.17.1
sphinx_markdown_tables==0.0.17
sphinx-argparse==0.4.0
sphinxcontrib.katex==0.9.6
sphinxext.opengraph==0.8.2
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.16.1
traitlets==5.10.0
nbsphinx==0.9.1
pandoc==2.3
pypandoc==1.11
GitPython==3.1.37
moto[s3]<5,>=4.0.1
mock-ssh-server==0.9.1
cryptography==41.0.4
pytest-httpserver<1.1,>=1.0.4
setuptools<=59.5.0

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

[health_checker]
pynvml<12,>=11.5.0

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

[mlflow]
mlflow<3.0,>=2.5.0

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

[nlp]
transformers<4.34,>=4.11
datasets<3,>=2.4

[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

[sentencepiece]
protobuf<3.21
sentencepiece==0.1.99

[slack]
slack_sdk<4,>=3.19.5

[streaming]
mosaicml-streaming<1.0
boto3<2,>=1.21.45
paramiko<3,>=2.11.0

[system_metrics_monitor]
pynvml<12,>=11.5.0

[tensorboard]
tensorboard<3.0.0,>=2.9.1

[timm]
timm<0.6,>=0.5.4

[unet]
monai<1.3,>=0.9.1
scikit-learn<2,>=1.0.1

[vit]
vit_pytorch==0.35.8

[wandb]
wandb<0.16,>=0.13.2
