pytorch-ndonnx

pytorch-ndonnx is a library that can be used to convert pytorch models to ONNX. It is based on ndonnx.

Conversion is achieved by ducktyping existing code using a pytorch_ndonnx.Tensor object.

See here for API documentation.

Installation

pip install pytorch-ndonnx

Coming soon

You can install the package in development mode using:

git clone https://github.com/cbourjau/pytorch-ndonnx
cd pytorch-ndonnx
pixi run pre-commit-install
pixi run postinstall
pixi run test