From PyTorch

About PyTorch

PyTorch is a machine learning framework based on the Torch library, used for applications such as computer vision and natural language processing, originally developed by Meta AI and now part of the Linux Foundation umbrella. It is free and open-source software released under the modified BSD license. Although the Python interface is more polished and the primary focus of development, PyTorch also has a C++ interface.

Model deployment from PyTorch

Model deployment from PyTorch is simple to achieve by exporting your PyTorch model to ONNX and subsequently using (if neccesary) the sclblonnx package to clean and check the resulting graph for an upload to the Nx AI cloud.

  • You can find details on PyTorch to ONNX exports here. You can find an insightful tutorial here.

  • You can find an example using PyTorch and sclblonnx here.

After obtaining a clean ONNX graph that adheres to our requirements, you can upload it to the Nx AI cloud for deployment.

Last updated