Crate axonml_serialize 

Axonml Serialize - Model Serialization for Axonml ML Framework

File

crates/axonml-serialize/src/lib.rs

Author

Andrew Jewell Sr - AutomataNexus

Updated

March 8, 2026

Disclaimer

Use at own risk. This software is provided “as is”, without warranty of any kind, express or implied. The author and AutomataNexus shall not be held liable for any damages arising from the use of this software.

Structs

Checkpoint

A complete training checkpoint.

CheckpointBuilder

Builder for creating checkpoints.

StateDict

State dictionary for storing model parameters.

StateDictEntry

An entry in the state dictionary.

TensorData

Serializable tensor data.

TrainingState

Training state for checkpointing.

Enums

Format

Supported serialization formats.

OnnxOpType

ONNX operator type mapping.

Functions

convert_from_pytorch

Convert a state dict from PyTorch naming conventions.

detect_format

Detect the format from a file path based on extension.

detect_format_from_bytes

Detect format from file contents (magic bytes).

from_onnx_shape

Convert from ONNX shape (handling -1 for dynamic dimensions).

from_pytorch_key

Convert a PyTorch-style key to Axonml format.

load_checkpoint

Load a training checkpoint.

load_state_dict

Load a state dictionary from a file.

pytorch_layer_mapping

Map of PyTorch layer names to Axonml equivalents.

save_checkpoint

Save a complete training checkpoint.

save_model

Save a model’s state dictionary to a file.

save_state_dict

Save a state dictionary to a file with specified format.

to_onnx_shape

Convert a shape to ONNX format (with batch dimension handling).

to_pytorch_key

Convert a Axonml key to PyTorch format.

transpose_linear_weights

Transpose weights if needed for format conversion.