ONNX¶
Open Neural Network Exchange (ONNX) is an open standard format for representing machine learning and deep learning models. It enables interoperability between different frameworks and streamlines the path from research to production.
The flytekit onnx type plugin comes in three flavors:
pip install flytekitplugins-onnxpytorch
This plugin enables the conversion from scikitlearn models to ONNX models.
pip install flytekitplugins-onnxtensorflow
This plugin enables the conversion from tensorflow models to ONNX models.
pip install flytekitplugins-onnxpytorch
This plugin enables the conversion from pytorch models to ONNX models.
Note
If you’d like to add support for a new framework, please create an issue and submit a pull request to the flytekit repo. You can find the ONNX plugin source code here.