Skip to main content

Module mamba2

burn_mamba

Module mamba2 

Source

Modules§

bidi
cache
Mamba-2 Inference Caches
layer
Mamba-2 Layer and Layer Stack
mamba2
Mamba-2 SSM Block — Structured State Space Duality (SSD)
network
Mamba-2 Language Model Network
prelude
ssd