Staff Software Engineer, GPU Performance, Core ML
Description For Staff Software Engineer, GPU Performance, Core ML
Google's Core Machine Learning organization is seeking a Staff Software Engineer to lead GPU performance optimization efforts. This role sits at the intersection of machine learning infrastructure and hardware acceleration, focusing on optimizing GPU performance for Google's ML workloads. The position requires deep expertise in both software development and machine learning, with a particular emphasis on GPU architectures and optimization techniques.
The role involves working with cutting-edge GPU technology to improve the performance of Google's machine learning systems that power services used by billions of users worldwide. You'll be responsible for optimizing the entire GPU software stack, from low-level kernels to high-level model design, while collaborating with experts across different domains.
As part of the ML, Systems, & Cloud AI organization, you'll contribute to the infrastructure that powers Google's core services and Cloud AI offerings, including Vertex AI platform. The position offers unique opportunities to work with the latest GPU technologies, sophisticated performance optimization tools, and influence the future of AI acceleration at Google.
The ideal candidate will bring together technical leadership, deep understanding of GPU architectures, and machine learning expertise to drive significant performance improvements across Google's ML infrastructure. This is a chance to shape the future of AI computing at one of the world's leading technology companies, with direct impact on products used by billions of users.
Responsibilities For Staff Software Engineer, GPU Performance, Core ML
- Build optimizations that improve benchmarks, but also power Google's most critical products and services
- Shape the entire GPU software stack through influencing model design, optimizing low-level kernels and compilers
- Resolve the most challenging performance bottlenecks and explore groundbreaking optimization techniques
- Collaborate with experts in ML, compiler design, and systems architecture through internal and external partnerships
Requirements For Staff Software Engineer, GPU Performance, Core ML
- Bachelor's degree or equivalent practical experience
- 8 years of experience in software development and with data structures/algorithms (e.g., C++ or Python)
- 5 years of experience with Machine Learning (ML) design and ML infrastructure
- Experience working with GPUs
- Experience in a technical leadership role leading project teams and setting technical direction
Benefits For Staff Software Engineer, GPU Performance, Core ML
- Comprehensive health benefits
- Retirement plans
- Parental leave
Jobs Related To Google Staff Software Engineer, GPU Performance, Core ML
