Performance Engineer, Kernels

Sarvam serves language, speech and multimodal models across a multi-node GPU fleet. You will author and optimize production GPU kernels and explain their performance impact.

• Author custom CUDA, DSL-based and PTX kernels
• Optimize kernels against stock libraries and published baselines
• Profile kernels with Nsight Compute and Nsight Systems
• Modify or author attention kernels for production workloads
• Diagnose and explain production p99 performance changes

Experience

5+ years

Domain

ai infrastructure

Sarvam has 28 open roles in Bangalore, at 732, Chinmaya Mission Hospital Road, Indiranagar Stage 1, Bengaluru, Karnataka 560038. See every Sarvam role.