Axel Feldmann

Axel Feldmann is a machine learning engineer on the GPU performance team at Jane Street. Prior to that, he did a PhD in Computer Science at MIT working on designing and programming hardware accelerators.
Avatar photo

Posts by Axel Feldmann

Developer Tools & Techniques

Accelerating Mamba-2 with High-Performance Forward Pass Kernels in NVIDIA CUDA Tile

NVIDIA CUDA Tile (cuTile) is a Python-based DSL and parallel programming model for NVIDIA GPUs that automatically leverages advanced hardware capabilities to... 15 MIN READ