Job Summary:
Google is a leading technology company that develops next-generation technologies impacting billions of users. They are seeking a Senior Software Engineer to optimize GPU performance for critical products, driving innovations in AI and accelerated computing.
Responsibilities:
• Build optimizations for the latest generation of GPUs that power Google’s most critical products and services, impacting billions of users worldwide.
• Address the most challenging performance bottlenecks through Google’s unparalleled access to the latest generation of GPUs, tooling, and a decade of experience building AI accelerators.
• Drive optimizations across Google’s GPU software stack from ML compiler cost model design to optimizing high performance GPU kernels to cross node model serving configurations.
• Influence the technical direction of the GPU software ecosystem at Google by collaborating with ML, compiler design, and systems architecture teams.
• Influence the deployment of Google’s GPU fleet by working with various product teams across Google.
Qualifications:
Required:
• Bachelor’s degree or equivalent practical experience.
• 5 years of experience with software development in one or more programming languages.
• 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.
• Experience with modern GPU architectures (NVIDIA, AMD, or other AI accelerators), memory hierarchies, and performance bottlenecks.
• Experience low-level GPU programming (CUDA, Triton, CUTLASS, etc.) and performance engineering techniques.
• Experience with modern LLMs and their deployment on AI accelerators.
Preferred:
• Master's degree or PhD in Computer Science or related technical field.
• 5 years of experience with data structures and algorithms.
• 1 year of experience in a technical leadership role.
• Experience with compiler optimization, code generation, and runtime systems for GPU architectures (OpenXLA, MLIR, Triton, etc.).
Company:
Google specializes in internet-related services and products, including search, advertising, and software. It is a sub-organization of Alphabet. Founded in 1998, the company is headquartered in Mountain View, USA, with a team of 10001+ employees. The company is currently Late Stage.