cloudpickle
ml_dtypes
numpy
packaging
psutil
scipy
tornado
typing_extensions

[all-prod]
cloudpickle
coremltools
ethos-u-vela
future
ml_dtypes
numpy
onnx
onnxoptimizer
onnxruntime
packaging
psutil
scipy
tensorflow
tensorflow-estimator
tflite
torch
torchvision
tornado
typing_extensions
xgboost>=1.1.0

[dev]
astroid
autodocsumm
black==20.8b1
commonmark>=0.7.3
cpplint
docutils<0.17
image
matplotlib
pillow
pylint
sphinx
sphinx_autodoc_annotation
sphinx_gallery
sphinx_rtd_theme
types-psutil

[importer-coreml]
coremltools

[importer-keras]
tensorflow
tensorflow-estimator

[importer-onnx]
future
onnx
onnxoptimizer
onnxruntime
torch
torchvision

[importer-pytorch]
future
torch
torchvision

[importer-tensorflow]
tensorflow
tensorflow-estimator

[importer-tflite]
tensorflow
tensorflow-estimator
tflite

[tvmc]
ethos-u-vela
future
onnx
onnxoptimizer
onnxruntime
tensorflow
tflite
torch
torchvision
xgboost>=1.1.0

[xgboost]
future
torch
xgboost>=1.1.0
