jamba

Model type

jamba

Class

JambaCausalLMModel

Task

hybrid-text-generation

Source

models/jamba.py

Description

Jamba hybrid SSM+Attention causal language model.

Usage

mobius build --model <MODEL_ID> output_dir/
from mobius import build

model = build("<MODEL_ID>")