Hugging Face's logo Hugging Face
  • Models
  • Datasets
  • Spaces
  • Docs
  • Enterprise
  • Pricing

  • Log In
  • Sign Up

kernels-community
/
mamba-ssm

kernel
Model card Files Files and versions
xet
Community
1
mamba-ssm / torch-ext /mamba_ssm /ops /triton
Ctrl+K
Ctrl+K
  • 4 contributors
History: 1 commit
danieldk's picture
danieldk HF Staff
Import mamba-ssm kernels
23d26f4 7 months ago
  • __init__.py
    0 Bytes
    Import mamba-ssm kernels 7 months ago
  • k_activations.py
    5.17 kB
    Import mamba-ssm kernels 7 months ago
  • layer_norm.py
    36.6 kB
    Import mamba-ssm kernels 7 months ago
  • layernorm_gated.py
    17.8 kB
    Import mamba-ssm kernels 7 months ago
  • selective_state_update.py
    12.1 kB
    Import mamba-ssm kernels 7 months ago
  • softplus.py
    328 Bytes
    Import mamba-ssm kernels 7 months ago
  • ssd_bmm.py
    13.6 kB
    Import mamba-ssm kernels 7 months ago
  • ssd_chunk_scan.py
    105 kB
    Import mamba-ssm kernels 7 months ago
  • ssd_chunk_state.py
    65.4 kB
    Import mamba-ssm kernels 7 months ago
  • ssd_combined.py
    62 kB
    Import mamba-ssm kernels 7 months ago
  • ssd_state_passing.py
    16.4 kB
    Import mamba-ssm kernels 7 months ago