PRF·0176
Software Engineer - GPU Kernels
Summary
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.
Responsibilities as stated
- 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.
Location, employment, and compensation
- Location. San Francisco, Toronto, New York, Montreal — hybrid
- Employment. unknown
- Compensation. Not stated on the source. NearMetal does not estimate compensation.
- How to apply. On the company’s own posting. NearMetal does not accept applications.
Why this is filed as systems software
- 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
This summary was written from the company’s own posting. NearMetal does not reproduce the full description and does not accept applications.