Views
No views yet
kernels library. This card was automatically generated.1# make sure `kernels` is installed: `pip install -U kernels`
2from kernels import get_kernel
3
4kernel_module = get_kernel("kernels-community/mamba-ssm", version=1)
5selective_scan_fn = kernel_module.selective_scan_fn
6
7selective_scan_fn(...)selective_scan_fnmamba_inner_fnfalcon_mamba_inner_fnselective_state_updatemamba_chunk_scan_combinedmamba_split_conv1d_scan_combinedMambaMamba2MambaLMHeadModel