# core.trainers.mamba – Axolotl

# core.trainers.mamba

`core.trainers.mamba`

Module for mamba trainer

## Classes

Name | Description
--- | ---
AxolotlMambaTrainer | Mamba specific trainer to handle loss calculation

### AxolotlMambaTrainer

    core.trainers.mamba.AxolotlMambaTrainer(
        *_args,
        bench_data_collator=None,
        eval_data_collator=None,
        dataset_tags=None,
        **kwargs,
    )

Mamba specific trainer to handle loss calculation