pyre-extensions
docstring-parser==0.8.1
importlib-metadata
pyyaml
docker
filelock
fsspec
urllib3<1.27,>=1.21.1
tabulate

[dev]
aiobotocore==2.1.0
ax-platform[mysql]==0.2.3
black==22.3.0
boto3==1.20.24
captum>=0.4.0
flake8==3.9.0
fsspec[s3]==2022.1.0
google-api-core
google-cloud-batch>=0.5.0
google-cloud-logging>=3.0.0
google-cloud-runtimeconfig>=0.33.2
hydra-core
ipython
kfp==1.8.9
moto==3.0.2
pyre-extensions==0.0.21
pytest
pytorch-lightning==1.5.10
torch-model-archiver>=0.4.2
torch>=1.10.0
torchmetrics<0.11.0
torchserve>=0.4.2
torchtext>=0.11.0
torchvision>=0.11.1
ts==0.5.1
usort==1.0.2
ray[default]

[gcp_batch]
google-cloud-batch>=0.5.0
google-cloud-logging>=3.0.0
google-cloud-runtimeconfig>=0.33.2

[kfp]
kfp==1.6.2

[kubernetes]
kubernetes>=11

[ray]
ray>=1.12.1
