Gpu Kernel Engineer • Mid-Level

Who you are

You have a strong background in low-level programming, particularly in C and C++, with a focus on optimizing performance for GPU workloads. Your experience with Linux environments allows you to navigate and troubleshoot complex systems effectively. You understand the intricacies of machine learning models and how to enhance their performance through kernel optimization. You thrive in collaborative settings, working alongside engineers who are equally passionate about pushing the boundaries of AI technology. Your problem-solving skills are top-notch, enabling you to tackle challenges that arise in high-impact systems work. You are eager to contribute to a team that values technical excellence and innovation.

Desirable

Experience with CUDA programming is a plus, as it will enhance your ability to optimize GPU kernels for machine learning applications. Familiarity with AI frameworks and libraries will also be beneficial in understanding the context of your work and its impact on model performance.

What you'll do

As a GPU Kernel Engineer at Baseten, you will be responsible for designing and implementing high-performance GPU kernels that power state-of-the-art machine learning models. You will collaborate with the Model Performance team to drive optimizations that enhance the efficiency of AI workloads. Your work will directly influence the performance of systems serving millions of users, making your contributions critical to the success of our platform. You will engage in code reviews and provide mentorship to junior engineers, fostering a culture of learning and growth within the team. Additionally, you will participate in cross-functional discussions to align engineering efforts with product goals, ensuring that the solutions you develop meet the needs of our users.

What we offer

At Baseten, we provide a dynamic work environment where innovation is encouraged and rewarded. You will have the opportunity to work on groundbreaking projects that shape the future of AI technology. We offer competitive compensation and benefits, along with a commitment to professional development and career growth. Our team is diverse and inclusive, and we believe that varied perspectives lead to better solutions. Join us in our mission to empower AI companies with the tools they need to succeed.