Crate ferrum_quantization 

Weight-format abstraction for Ferrum models.

Separates “what is the weight matrix like” (dense f32, GPTQ int4, AWQ, GGUF, …) from “what device does the math” (Backend) and “how does the model wire things together” (model code).

Usage in model code:

let qkv: Box<dyn Linear<B>> = loader.load_linear("model.layers.0.self_attn.qkv_proj")?;
qkv.forward(ctx, &input, &mut out, m);

The Linear trait dispatches to the appropriate backend kernel (B::gemm for Dense, B::gemm_gptq for GPTQ, etc.) without the model having to branch on quantization type.

Re-exports

pub use block_fp8_safetensors_source::BlockFp8SafetensorsSource;

pub use block_fp8_safetensors_source::BLOCK_FP8_E4M3_SOURCE_FORMAT_ID;

pub use compressed_tensors_marlin_source::CompressedTensorsMarlinSafetensorsSource;

pub use compressed_tensors_marlin_source::COMPRESSED_TENSORS_MARLIN_INT4_FORMAT_ID;

pub use compressed_tensors_marlin_source::COMPRESSED_TENSORS_MARLIN_INT4_SYMMETRIC_FORMAT_ID;

pub use dense::DenseLinear;

pub use gguf::GgufFile;

pub use gguf::GgufLinear;

pub use gguf::GgufLoader;

pub use gguf::GgufWeightComponentSource;

pub use gptq::GptqLinear;

pub use gptq::StackedExpertLinear;

pub use gptq_marlin_source::GptqMarlinSafetensorsSource;

pub use gptq_marlin_source::GPTQ_MARLIN_INT4_FORMAT_ID;

pub use loader::PrefixedLoader;

pub use loader::WeightLoader;

pub use lora::LoraLinearRef;

pub use mxfp4_safetensors_source::Mxfp4SafetensorsSource;

pub use mxfp4_safetensors_source::MXFP4_E2M1_E8M0_SOURCE_FORMAT_ID;

pub use native_safetensors::NativeSafetensorsLoader;

pub use quant_linear::QuantLinear;

pub use safetensors_archive::SafetensorsArchive;

pub use safetensors_archive::SafetensorsTensor;

pub use config::QuantConfig;

pub use config::QuantMethod;

Modules

block_fp8_safetensors_source

Exact safetensors source adapter for rectangular block-FP8 checkpoints.

compressed_tensors_marlin_source

Exact compressed-tensors W4 adapter for the Marlin physical ABI.

config

Quantization configuration parsed from model metadata.

dense

Dense linear projection — the baseline, uses B::gemm directly.

gguf

GGUF (GGML Universal Format) reader.

gptq

GPTQ linear projection — thin factory wrapper.

gptq_marlin_source

Typed safetensors GPTQ adapter for the Marlin physical ABI.

loader

WeightLoader trait — unified interface for loading tensor/linear weights into a specific backend.

lora

LoRA reference utilities.

mxfp4_safetensors_source

Exact source adapter for GPT-OSS safetensors MXFP4 expert weights.

native_safetensors

Native safetensors WeightLoader<B> — mmap + safetensors crate, no candle dependency on the LLM hot path.

quant_linear

QuantLinear<B> — thin wrapper that delegates to the boxed Linear<B> returned by B::load_quant / B::load_quant_fused.

safetensors_archive

traits

Re-export of Linear trait (canonical home: ferrum-kernels).

Traits

Linear

A weight-bearing linear projection.