Hi all,
I took a look around the Axon and Nx codebases trying to determine how a model can be saved to a file and later imported.
I am referring to this behavior:
https://pytorch.org/tutorials/beginner/saving_loading_models.html
My intended use case is to train the model locally, export the model to a file, and then upload it to the production server that can then use it (no training in the cloud).
I understand Axon/Nx are in their early days so just wondering if this can be done yet, if so, how?
(If I had to guess this should be a priority but I’m not sure where it falls on that immediate list.)