mamba-rs 0.2.0

Mamba SSM (Selective State Space Model) in Rust with CUDA GPU support. Training + inference, forward + backward (BPTT), burn-in, custom CUDA kernels.
Documentation 
1
2
3
4
5
6
7
8

//! Shared operations: dimensions, BLAS, math, normalization utilities.

pub mod blas;
pub mod dims;
pub mod fast_math;
pub mod norms;

pub use dims::{MambaDims, MambaRecurrentState};