Trait Module 

pub trait Module<R: Runtime> {
    // Required methods
    fn parameters(&self) -> Vec<&Var<R>>;
    fn named_parameters(&self) -> Vec<(String, &Var<R>)>;

    // Provided method
    fn num_parameters(&self) -> usize { ... }
}

Core trait for neural network modules.

Provides access to trainable parameters for optimizers and named parameters for checkpoint serialization.

Forward passes stay as inherent methods on each layer because signatures differ (different client bounds, input types).

Required Methods

fn parameters(&self) -> Vec<&Var<R>>

All trainable parameters (for optimizer).

fn named_parameters(&self) -> Vec<(String, &Var<R>)>

Named parameters (for checkpointing). Names use dot notation for nested modules: "layers.0.attn.weight".

Provided Methods

fn num_parameters(&self) -> usize

Total number of scalar parameters.

Implementors