Skip to content

Mamba inference opt - #4414

Merged
wdykas merged 5 commits into
NVIDIA:mainfrom
wdykas:mamba-inference-opt
Apr 22, 2026
Merged

Mamba inference opt#4414
wdykas merged 5 commits into
NVIDIA:mainfrom
wdykas:mamba-inference-opt

Commits

Commits on Apr 21, 2026

  • no recompute

    rootroot
    root
    authored and
    root
    committed
    a0dda2c

  • optimization mamaba

    rootroot
    root
    authored and
    root
    committed
    bf56d3d

Commits on Apr 22, 2026

  • revert

    author
    William Dykas
    committed
    678b590

  • top sorted only during training

    author
    William Dykas
    committed
    b67fd23

  • fix cuda graphs

    rootroot
    root
    authored and
    root
    committed
    8d27f1d