How to save and load models using Axon/Nx

There is GitHub - mortont/axon_onnx: Easily convert models between ONNX and Axon · GitHub but it is early on.