# mamba2

| | |
|---|---|
| **Model type** | `mamba2` |
| **Class** | `Mamba2CausalLMModel` |
| **Task** | `ssm2-text-generation` |
| **Source** | `models/mamba.py` |

## Description

Mamba2/SSD causal language model with multi-head SSM.

## Usage

```bash
mobius build --model <MODEL_ID> output_dir/
```

```python
from mobius import build

model = build("<MODEL_ID>")
```