About the job
ABOUT BASETEN
At Baseten, we empower the world's leading AI firms—such as Cursor, Notion, and OpenEvidence—by delivering mission-critical inference solutions. Our unique blend of applied AI research, robust infrastructure, and user-friendly developer tools enables AI pioneers to effectively deploy groundbreaking models. With our recent achievement of a $300M Series E funding round supported by esteemed investors like BOND and IVP, we're on an exciting growth trajectory. Join our dynamic team and contribute to the platform that drives the next generation of AI products.
THE ROLE
We are looking for an experienced Senior GPU Kernel Engineer to join our innovative team at the forefront of AI acceleration. In this role, your programming expertise will directly enhance the performance of cutting-edge machine learning models. You'll be responsible for developing highly efficient GPU kernels that optimize computational processes, allowing for transformative AI applications.
You'll thrive in a fast-paced, intellectually challenging environment where your technical skills are pivotal. Your contributions will directly affect production systems that serve millions of users across various platforms. This position offers exceptional opportunities for career advancement for engineers enthusiastic about low-level optimization and impactful systems engineering.
EXAMPLE INITIATIVES
As part of our Model Performance team, you will engage in projects like:
RESPONSIBILITIES
Core Engineering Responsibilities
Design and develop high-performance GPU kernels for essential machine learning operations, including matrix multiplications and attention mechanisms.
Collaborate with cross-functional teams to drive performance improvements and implement optimizations.
Debug and refine kernel code to achieve maximal efficiency and reliability.
Stay abreast of the latest advancements in GPU technology and machine learning frameworks.

