Senior AI/ML Performance Engineer

Requirements:

• Bachelor’s degree or equivalent practical experience.
• 5 years of experience with software development in one or more programming languages.
• 3 years of experience with one or more of the following: Speech/audio, reinforcement learning, ML infrastructure, or specialization in another ML field.
• 3 years of experience with ML infrastructure.
• Experience with AI and agentic tooling for development and research.

Nice to haves:

• Master's degree or PhD.
• 5 years of experience with data structures/algorithms in C++ and Python.
• Experience with algorithms, systems and tools for ML performance projections and evaluation.
• Experience designing or implementing components of a Deep Learning Compiler Stack.
• Experience in low-latency systems programming and optimizing data movement.
• Experience in performance engineering for ML/AI, including optimization of GPU/TPU kernels.

What you'll be doing:

• Design and implement solutions in specialized ML areas.
• Build, maintain, and validate HW/SW tooling for ML/AI infrastructure.
• Measure, visualize, and analyze ML HW/SW performance.
• Define, implement, and validate performance and cost metrics for workloads and systems.
• Collaborate with other teams to improve the end-to-end flow and results.

Perks and benefits:

• Joining Google's innovative software engineering team.
• Opportunities for growth and project flexibility.
• Work on cutting-edge technologies at a global scale.
• Contribute to advanced AI and agentic approaches to HW/SW co-design.
• Being part of the MSCA organization focused on ML, Systems, and Cloud AI.