Expand description
2026-09-25: FP8 block-scaled weight loading, BF16-to-NVFP4 quantization, and the attention, SSM and MoE layer loaders.
Owner: model-layers (weight loading). Invariants:
- Every
Fp8Weightreturned here holds its block scale as an FP32 device buffer.
Functionsยง
- load_
attention - 2026-09-25: Load the
{layer_prefix}.self_attnweights of a full-attention layer. - load_
fp8_ block_ scaled_ as_ fp8weight - 2026-09-25: Load an FP8 E4M3 block-scaled checkpoint weight as a native
Fp8Weight. - load_
moe - 2026-09-25: Load the
{layer_prefix}.mlpMoE weights of a layer. - load_
moe_ skip_ experts - 2026-09-25:
load_moewith every routed expert left asExpertWeight::null(). The native-FP8 qwen3 loader uses it and loads the routed experts separately. - load_
ssm - 2026-09-25: Load the
{layer_prefix}.linear_attnweights of a linear-attention layer. - quantize_
to_ nvfp4 - 2026-09-25: Quantize a BF16 dense
[n, k]weight to NVFP4 on the GPU.