Software Engineer - GPU Kernels

About baseten

Baseten powers inference for AI companies.

What you'd do

• Design and implement high-performance GPU kernels
• Optimize code using CUDA and PTX
• Identify and resolve performance bottlenecks
• Contribute to internal and open-source GPU libraries

What they're looking for

• Strong understanding of GPU architecture
• Proficient in C++ and performance profiling tools
• Knowledge of CUDA C++ API
• Understanding of numerical precision strategies

Perks

• 100% coverage of medical, dental, and vision for員
• Flexible PTO policy including winter break
• Paid parental leave