mamba-ssm / README.md
drbh's picture
drbh HF Staff
feat: add tag for hfjob build
f77c224 verified
metadata
license: apache-2.0
tags:
  - kernel

Status

Mamba

Mamba state space kernels + models from state-spaces/mamba.

Warning

Some functionality is dependent on einops and transformers, however we currently don't have any way of defining these dependencies yet. The scope of the Hub kernel is probably too large (should maybe only contain the selective-scan and Triton kernels).