NVIDIAHow-ToDevelopersOctober 19, 2026

NVIDIA and Jane Street optimize Mamba-2 kernels in cuTile

Read original source →developer.nvidia.com

Jane Street ML engineers implemented Mamba-2's forward pass in NVIDIA's cuTile, a Python DSL for tile-level GPU programming that targets Tensor Cores automatically. The post walks through the three custom kernels behind the forward pass and the cuTile optimizations used.

1 source

More stories today

Open the live feed