Software Engineering • Mid-Level

Who you are

You have a Bachelor's degree or equivalent practical experience, along with at least 2 years of experience in software development using one or more programming languages. Your background includes low-level GPU programming with technologies such as CUDA, Triton, and CUTLASS, as well as performance engineering techniques. You are familiar with modern GPU architectures, including NVIDIA and AMD, and understand memory hierarchies and performance bottlenecks. Ideally, you hold a Master's degree or PhD in Computer Science or a related technical field, and have 2 years of experience with data structures and algorithms in either an academic or industry setting.

You possess experience with compiler optimization, code generation, and runtime systems for GPU architectures, particularly with tools like OpenXLA and MLIR. Your understanding of modern Large Language Models (LLMs) and their deployment on AI accelerators is a significant asset. You thrive in collaborative environments and are eager to influence the deployment of Google’s GPU fleet by working with various product teams across the organization.

What you'll do

As a Software Engineer focused on GPU Performance, you will work on critical projects that enhance Google’s GPU software stack. Your responsibilities will include optimizing high-performance GPU kernels and contributing to the design of ML compiler cost models. You will collaborate with teams across Google, including ML, compiler design, and systems architecture, to influence the technical direction of the GPU software ecosystem. Your role will involve tackling performance bottlenecks and ensuring that Google’s products can handle information at massive scale.

You will have opportunities to switch teams and projects, allowing you to grow your skills and impact across the organization. Your contributions will directly affect how billions of users connect, explore, and interact with information and one another. You will be part of a team that values fresh ideas from various areas, including information retrieval, distributed computing, and artificial intelligence.

What we offer

At Google, you will be part of a culture that encourages innovation and collaboration. We provide competitive compensation and benefits, along with opportunities for professional growth and development. You will work in an environment that values diversity and inclusion, and where your contributions will have a meaningful impact on the technology that shapes the future. We encourage you to apply even if your experience doesn't match every requirement, as we believe in the potential of diverse teams to build better products.