This plugin allows you to generate ONNX models from your PyTorch models.

To install the plugin, run the following command:

```
pip install flytekitplugins-onnxpytorch
```

---
**Source**: https://github.com/unionai/unionai-docs/blob/main/content/api-reference/plugins/onnx-pytorch/_index.md
**HTML**: https://www.union.ai/docs/v1/flyte/api-reference/plugins/onnx-pytorch/