burn_mamba/modules/misc/mod.rs
1/// Group→head expansion of B/C (GQA-style sharing).
2pub mod gqa;
3pub mod rope;
4/// Optional `NaN`/`Inf` guards gated by [`crate::DENY_NAN`] / [`crate::DENY_INF`].
5pub mod sanity;
6/// Stable segment-sum → 1-semiseparable mask (log-space prefix-sum differences).
7pub mod segsum;
8/// Typed-array variant of `split_with_sizes` for clean destructuring.
9pub mod split;