Glm5NextKdaWeights

metrale_model_arch::glm5next_kda

Struct Glm5NextKdaWeights

pub struct Glm5NextKdaWeights {Show 13 fields
    pub q_proj: DenseWeight,
    pub k_proj: DenseWeight,
    pub v_proj: DenseWeight,
    pub conv: DenseWeight,
    pub f_a: DenseWeight,
    pub f_b: DenseWeight,
    pub dt_bias: DevicePtr,
    pub a_log: DevicePtr,
    pub b_proj: DenseWeight,
    pub g_a: DenseWeight,
    pub g_b: DenseWeight,
    pub o_norm: DenseWeight,
    pub o_proj: DenseWeight,
}

Expand description

2026-09-25: One KDA block’s device weights. Torch Linear layout [out, in], BF16, except the two F32 gate parameters. There is no Z tensor; the output gate is low-rank g_a/g_b.

Fields§

§q_proj: DenseWeight§k_proj: DenseWeight§v_proj: DenseWeight§conv: DenseWeight

2026-09-25: [conv_dim, conv_kernel] BF16: the q, k and v conv tensors concatenated, with the singleton middle dimension dropped.

§f_a: DenseWeight§f_b: DenseWeight§dt_bias: DevicePtr

2026-09-25: [heads * head_dim] F32, one value per channel.

§a_log: DevicePtr

2026-09-25: [heads] F32, one value per head (unlike dt_bias).

§b_proj: DenseWeight§g_a: DenseWeight§g_b: DenseWeight§o_norm: DenseWeight

2026-09-25: [head_dim] BF16.

§o_proj: DenseWeight

Auto Trait Implementations§

impl Freeze for Glm5NextKdaWeights

impl RefUnwindSafe for Glm5NextKdaWeights

impl Send for Glm5NextKdaWeights

impl Sync for Glm5NextKdaWeights

impl Unpin for Glm5NextKdaWeights

impl UnwindSafe for Glm5NextKdaWeights

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,