Software Engineer - GPU Kernels

Baseten · San Francisco, Toronto, New York, Montreal

Baseten is hiring a Software Engineer - GPU Kernels in San Francisco. The role focuses on operating systems & kernel, networking, graphics, gpu & compute, low-latency & performance.

Key responsibilities

• Design and implement high-performance GPU kernels for key ML operations, including matrix multiplications, attention mechanisms, and mixture-of-experts routing.
• Write and optimize code using CUDA, PTX assembly, and architecture-specific techniques.
• Contribute to internal and open-source GPU libraries.

Source and corrections

NearMetal classified and summarized this role from an official company source.