Join Google as a Staff Machine Learning Performance Engineer

We are seeking an experienced Staff Machine Learning Performance Engineer to join our dynamic team at Google. This role is crucial for driving cutting-edge performance and optimizing machine learning and AI training workloads across our vast technology landscape.

Minimum Qualifications:

• Bachelor’s degree or equivalent practical experience.
• 8 years of experience in software development, along with a strong understanding of data structures and algorithms.
• 5 years of proven track record in testing and launching software products, along with 3 years dedicated to software design and architecture.
• Extensive experience (5 years) with machine learning algorithms and tools such as TensorFlow, alongside expertise in artificial intelligence, deep learning, and natural language processing.

Preferred Qualifications:

• Experience in a technical leadership role with responsibilities in leading project teams and setting technical direction.
• Strong background in working within a complex, matrixed organization on cross-functional and/or cross-business projects.
• Skills in performance analysis and optimization, including system architecture and performance modeling.
• Knowledge in compiler optimizations or related fields.
• Experience in distributed development and large-scale data processing.

About the Role:

At Google, we develop next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our software engineers are tasked with solving diverse challenges across the full-stack, extending well beyond ordinary web search. This role offers the opportunity to work on impactful projects critical to Google’s ongoing needs, with the flexibility to switch focus as our business and technology evolve.

The TPU Performance team, which you will be joining, is at the forefront of enhancing machine learning and AI training workload efficiencies at scale. Our efforts significantly contribute to Google’s leadership in global technology, addressing evolving business challenges and opportunities in over 200 countries and territories.

Responsibilities:

• Lead efforts in optimizing large language models like Google Deepmind Gemini, Bard, Search Magi, and Cloud LLM APIs.
• Establish and maintain training and serving benchmarks that are central to Google production, across the industry and the ML community.
• Work closely with Google product teams to address and solve performance issues, ensuring efficient training and deployment of new models and technologies.
• Pioneer model/data efficiency techniques to enhance machine learning tasks and reduce data requirements.

This position offers a unique opportunity to work at the forefront of technology innovation alongside top engineers and leaders. If you are passionate about massive scale machine learning and leading performance optimization initiatives, we would love to hear from you.

Additional Information:

Company Name: Google
Job Title: Staff Machine Learning Performance Engineer

We encourage interested and qualified candidates to apply promptly as we aim to fill this vital role swiftly.

Google is proud to be an equal opportunity workplace committed to building a diverse team and fostering an inclusive environment for all employees.