Asaf Joseph Gardinandasafg a6c050286a [v1][mamba] Added mamba_type into MambaSpec (#21715) ...

Signed-off-by: asafg <asafg@ai21.com>
Co-authored-by: asafg <asafg@ai21.com>

2025-07-28 08:15:55 +00:00

..

ops

[Model] Replace Mamba2 RMSNorm Gated with Fused Triton Kernel (#20839)

2025-07-25 06:49:36 -07:00

__init__.py

[Kernel/Model] Migrate mamba_ssm and causal_conv1d kernels to vLLM (#7651)

2024-08-28 15:06:52 -07:00

abstract.py

[v1][mamba] Added mamba_type into MambaSpec (#21715)

2025-07-28 08:15:55 +00:00

mamba2_metadata.py

[Kernel] Triton implementation of causal-conv1d for Mamba-based models (#18218)

2025-07-09 12:53:55 -07:00

mamba_mixer2.py

[v1][mamba] Added mamba_type into MambaSpec (#21715)

2025-07-28 08:15:55 +00:00

mamba_mixer.py

[Kernel] Triton implementation of causal-conv1d for Mamba-based models (#18218)

2025-07-09 12:53:55 -07:00

mamba_utils.py

[V1] [Hybrid] Refactor mamba state shape calculation; enable V1 via cli (#20840)

2025-07-15 04:04:35 -07:00