metrale_model_arch/
lib.rs

1// SPDX-License-Identifier: MIT OR Apache-2.0
2
3//! 2026-09-25: Per-family model architectures, their weight loaders (`weight_loader`, `mistral_loader`) and the DFlash and MTP draft heads.
4//!
5//! Owner: model-arch.
6//! Invariants: none beyond the types.
7
8pub mod attn_v41;
9pub mod deepseek_v41_layer;
10pub mod deepseek_v41_ref;
11pub mod deepseek_v4_mtp;
12pub mod dflash_head;
13pub mod engram_v41;
14pub mod glm5next_dsa;
15pub mod glm5next_dsa_ref;
16pub mod glm5next_kda;
17pub mod glm5next_kda_ref;
18pub mod glm5next_layer;
19pub mod glm5next_mhc;
20pub mod glm5next_mlp;
21pub mod glm5next_mtp_head;
22pub mod glm5next_skeleton;
23pub mod kimi_k3;
24pub mod mistral_loader;
25pub mod moe_v41;
26pub mod nemotron_mamba2;
27pub mod nemotron_moe;
28#[cfg(test)]
29pub mod ple_tests;
30pub mod precision_schedule;
31pub mod seq_state_reserve;
32pub mod tp_shard;
33pub mod weight_loader;