Serialization Functionality

[seanmor5]

Checklist for serialization support of Axon models.

Linear

• [x] dense
• [ ] bilinear
• [ ] embedding

Convolutional

• [x] conv
• [ ] conv_transpose
• [ ] depthwise_conv
• [ ] separable_conv2d
• [ ] separable_conv3d

Dropout

• [ ] dropout
• [ ] spatial_dropout
• [ ] alpha_dropout
• [ ] feature_alpha_dropout

Pooling

• [ ] adaptive_avg_pool
• [ ] adaptive_max_pool
• [ ] adaptive_lp_pool
• [x] avg_pool
• [x] max_pool
• [x] lp_pool
• [x] global_avg_pool
• [x] global_max_pool
• [x] global_lp_pool

Normalization

• [ ] batch_norm
• [ ] layer_norm
• [ ] instance_norm
• [ ] group_norm

Recurrent

• [ ] conv_lstm
• [ ] lstm
• [ ] gru

Combinators

• [ ] concatenate
• [ ] add
• [ ] subtract
• [ ] multiply

Activations

• [x] celu
• [x] elu
• [x] exp
• [ ] gelu
• [ ] hard_sigmoid
• [ ] hard_silu
• [ ] hard_tanh
• [x] leaky_relu
• [x] linear
• [ ] log_sigmoid
• [ ] mish
• [x] relu
• [ ] relu6
• [x] selu
• [x] sigmoid
• [ ] silu
• [x] softmax
• [x] softplus
• [x] softsign
• [x] tanh

Special

• [ ] constant
• [ ] nx
• [ ] layer