pyyaml<7,>=6.0
tqdm<5,>=4.62.3
torchmetrics<0.8,>=0.7.0
torch_optimizer<0.4,>=0.3.0
torchvision>=0.10.0
torch<2,>=1.10
yahp<0.2,>=0.1.3
requests<3,>=2.26.0
numpy<1.23.0,>=1.21.5
psutil<6,>=5.8.0
coolname<2,>=1.1.0
tabulate==0.8.10
py-cpuinfo<9,>=8.0.0
packaging<22,>=21.3.0
importlib-metadata<5,>=4.11.0

[all]
docutils==0.17.1
pycocotools<3,>=2.0.4
recommonmark==0.7.1
comet_ml<4.0.0,>=3.31.12
yamllint==1.28.0
tensorboard<3.0.0,>=2.9.1
mosaicml-streaming
testbook==0.4.2
monai<0.10,>=0.9.1
pre-commit<3,>=2.18.1
tensorflow-io<0.28,>=0.26.0
paramiko<3,>=2.11.0
pandoc==2.2
mock-ssh-server==0.9.1
cryptography==38.0.1
sphinx-argparse==0.3.1
moto[s3]<5,>=4.0.1
datasets<3,>=2.4.0
sphinxcontrib-images==0.9.4
GitPython==3.1.27
onnx<2,>=1.12.0
ipykernel==6.15.3
nbsphinx==0.8.9
pytest==7.1.3
myst-parser==0.16.1
pypandoc==1.9
traitlets==5.4.0
jupyter==1.0.0
pytest_codeblocks==0.16.1
onnxruntime<2,>=1.12.1
vit_pytorch==0.35.8
transformers<5,>=4.11
wandb<0.14,>=0.13.2
custom_inherit==2.4.0
setuptools<=59.5.0
furo==2022.9.15
scikit-learn<2,>=1.0.1
timm<0.6,>=0.5.4
sphinx-copybutton==0.5.0
pytest-httpserver<1.1,>=1.0.4
sphinxcontrib.katex==0.9.0
sphinx==4.4.0
coverage[toml]==6.4.4
sphinx_markdown_tables==0.0.17
junitparser==2.8.0
sphinxext.opengraph==0.6.3
toml==0.10.2
sphinx_panels==0.6.0
boto3<2,>=1.21.45
deepspeed==0.5.10
apache-libcloud<4,>=3.3.1
fasteners==0.18
ipython==7.32.0
py-cpuinfo<9,>=8.0.0
sphinxemoji==0.2.0

[base]

[coco]
pycocotools<3,>=2.0.4

[comet_ml]
comet_ml<4.0.0,>=3.31.12

[deepspeed]
deepspeed==0.5.10

[dev]
custom_inherit==2.4.0
junitparser==2.8.0
coverage[toml]==6.4.4
fasteners==0.18
pytest==7.1.3
toml==0.10.2
ipython==7.32.0
ipykernel==6.15.3
jupyter==1.0.0
yamllint==1.28.0
recommonmark==0.7.1
sphinx==4.4.0
pre-commit<3,>=2.18.1
docutils==0.17.1
sphinx_markdown_tables==0.0.17
sphinx-argparse==0.3.1
sphinxcontrib.katex==0.9.0
sphinxext.opengraph==0.6.3
sphinxemoji==0.2.0
furo==2022.9.15
sphinx-copybutton==0.5.0
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.4.0
nbsphinx==0.8.9
pandoc==2.2
pypandoc==1.9
GitPython==3.1.27
moto[s3]<5,>=4.0.1
mock-ssh-server==0.9.1
cryptography==38.0.1
pytest-httpserver<1.1,>=1.0.4
setuptools<=59.5.0

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

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

[nlp]
transformers<5,>=4.11
datasets<3,>=2.4.0

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

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

[tensorboard]
tensorboard<3.0.0,>=2.9.1
tensorflow-io<0.28,>=0.26.0

[timm]
timm<0.6,>=0.5.4

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

[vit]
vit_pytorch==0.35.8

[wandb]
wandb<0.14,>=0.13.2
