mamba-rs 0.1.4

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

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

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

pub use dims::{MambaDims, MambaRecurrentState};