Software Engineer, ML Compiler

Google is a global technology leader providing cloud computing, search, and digital services to billions of users.
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
2+ years of experience
AI · Enterprise SaaS · Cloud

Description For Software Engineer, ML Compiler

Google Cloud's software engineers develop next-generation technologies that transform how billions of users connect, explore, and interact with information. This ML Compiler Engineer role focuses on developing and optimizing machine learning infrastructure within Google Cloud Platform. You'll work on critical projects involving compiler optimization, ML model performance analysis, and code generation for GPU compilers. The role offers opportunities to work with cutting-edge ML infrastructure, contribute to Google's planet-wide platform, and solve challenging problems in machine learning compilation.

As part of Google Cloud, you'll be working on technologies that serve diverse clients from startups to billion-user companies, providing advanced data and analytics solutions. The position requires expertise in ML infrastructure, compiler optimization, and modern C++ development. You'll collaborate with teams across Google to improve code generation, analyze model performance, and integrate new technologies like MLIR.

The role offers growth opportunities through various rotations, including bug triaging and guiding external contributions. You'll be empowered to act like an owner, innovate, and tackle complex technical challenges. Google's commitment to diversity, equality, and inclusion ensures a supportive work environment where you can make a significant impact on the future of cloud computing and machine learning infrastructure.

Working at Google Cloud means being part of a team that's pushing the boundaries of technology while maintaining high standards of security and performance. You'll have the chance to work with state-of-the-art tools and technologies while contributing to solutions that help businesses worldwide harness the power of Google's infrastructure.

Last updated 8 days ago

Responsibilities For Software Engineer, ML Compiler

  • Profile and analyze the performance of compiled ML models
  • Improve code generation in the XLA GPU compiler or the Triton compiler
  • Serve on rotations that triage bugs, guide Public Relations from external contributors, or integrate Multi-Level Intermediate Representation (MLIR) into Google

Requirements For Software Engineer, ML Compiler

  • Bachelor's degree in Computer Science, related technical/quantitative field, or equivalent practical experience
  • 2 years of experience with software development in one or more programming languages, and with data structures/algorithms
  • 2 years of experience with ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging)
  • Experience in developing software systems in modern C++
  • 3 years of experience working on compiler optimization (preferred)

Interested in this job?

Jobs Related To Google Software Engineer, ML Compiler

Research Scientist, Geo Algorithms, Research

Research Scientist position at Google focusing on Geo Algorithms research, requiring PhD and programming expertise, offering competitive compensation and research opportunities.

Software Developer III, Machine Learning, Google Cloud AI

Software Developer III position at Google Cloud AI, focusing on machine learning development and implementation with 2+ years of experience required.

Research Scientist, Gemini

Research Scientist position at Google working on Gemini AI, combining advanced machine learning research with practical implementation to develop innovative AI solutions.

Research Scientist, Google Cloud AI

Research Scientist position at Google Cloud AI, focusing on advancing AI technology through fundamental research and product innovation.

Natural Language Processing Research Scientist

Research Scientist position at Google focusing on improving factuality and grounding in Large Language Models, requiring PhD in Computer Science and NLP expertise.