burn_mamba/mamba3/single_ssd/ssd/serial_recalculated/mod.rs
1//! Single-SSD serial scan with a custom, memory-efficient backward.
2//!
3//! The forward + [`Mamba3SingleSsdBackendExt`] impl live in
4//! `serial_recalculated`; the registered autodiff [`backward`](crate::mamba3::single_ssd::ssd::serial_recalculated::backward) node and the
5//! recompute-based gradient math in [`combined_backward`](crate::mamba3::single_ssd::ssd::serial_recalculated::combined_backward) save training memory
6//! by recomputing intermediates instead of storing them.
7
8/// The registered custom `Backward` node (autodiff op).
9#[cfg(feature = "autodiff")]
10pub mod backward;
11/// Recompute-based gradient math (the memory-efficient backward).
12pub mod combined_backward;
13/// The same-step γ-correction on primitives — forward and analytic backward.
14pub mod diag;
15mod serial_recalculated;
16
17pub use serial_recalculated::Mamba3SingleSsdBackendExt;
18
19#[cfg(feature = "autodiff")]
20pub use serial_recalculated::Mamba3SingleSsdAutodiffBackendExt;