NemotronSsmWeights

metrale_model_layers::weight_map::nemotron

Struct NemotronSsmWeights

pub struct NemotronSsmWeights {
    pub in_proj: QuantizedWeight,
    pub out_proj: QuantizedWeight,
    pub conv1d_weight: DenseWeight,
    pub conv1d_bias: DenseWeight,
    pub a_log: DenseWeight,
    pub d_param: DenseWeight,
    pub dt_bias: DenseWeight,
    pub ssm_norm: DenseWeight,
}

Expand description

2026-09-25: Nemotron-H Mamba-2 mixer weights.

in_proj produces [z(d_inner), x(d_inner), B(n_groups*state), C(n_groups*state), dt(num_heads)].

Fields§

§in_proj: QuantizedWeight

2026-09-25: NVFP4, or QuantizedWeight::null() when the layer is not NVFP4.

§out_proj: QuantizedWeight

2026-09-25: NVFP4, or QuantizedWeight::null() when the layer is not NVFP4.

§conv1d_weight: DenseWeight§conv1d_bias: DenseWeight

2026-09-25: F32, widened from the BF16 checkpoint tensor at load.

§a_log: DenseWeight

2026-09-25: F32, widened from the BF16 checkpoint tensor at load.

§d_param: DenseWeight

2026-09-25: The D skip connection; F32, widened from BF16 at load.

§dt_bias: DenseWeight

2026-09-25: F32, widened from the BF16 checkpoint tensor at load.

§ssm_norm: DenseWeight

Auto Trait Implementations§

impl Freeze for NemotronSsmWeights

impl RefUnwindSafe for NemotronSsmWeights

impl Send for NemotronSsmWeights

impl Sync for NemotronSsmWeights

impl Unpin for NemotronSsmWeights

impl UnwindSafe for NemotronSsmWeights

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T> Instrument for T

fn instrument(self, span: Span) -> Instrumented<Self>

Instruments this type with the provided [Span], returning an Instrumented wrapper. Read more

fn in_current_span(self) -> Instrumented<Self>

Instruments this type with the current Span, returning an Instrumented wrapper. Read more

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T> Same for T

type Output = T

Should always be Self

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

fn vzip(self) -> V

impl<T> WithSubscriber for T

fn with_subscriber<S>(self, subscriber: S) -> WithDispatch<Self>
where S: Into<Dispatch>,

Attaches the provided Subscriber to this type, returning a [WithDispatch] wrapper. Read more

fn with_current_subscriber(self) -> WithDispatch<Self>

Attaches the current default Subscriber to this type, returning a [WithDispatch] wrapper. Read more

impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
where ST: ?Sized, DT: ?Sized,

impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
where ST: ?Sized, DT: ?Sized,

impl<T> Read<Exclusive, BecauseExclusive> for T
where T: ?Sized,