Senior Software Engineer, ML Performance Infrastructure

Google is a leading technology company that develops next-generation technologies impacting billions of users. They are seeking a Senior Software Engineer to contribute to machine learning infrastructure, optimize performance, and collaborate across teams to enhance AI capabilities.

Responsibilities:

• Support new and exciting Machine Learning (ML) paradigms, such as horizontal scaling for upcoming TPU chips, by making contributions across the end-to-end stack and analysis tools
• Learn and build an intuitive understanding of existing data collection, analysis, and visualization workflows
• Understand model optimization use-cases, drive cross-functional efforts to deliver on chip-profiling requirements, and propose new hardware features
• Collaborate across hardware, driver, runtime, and performance analysis teams and other key stakeholders
• Collect and analyze profile data to provide expert-level visualizations and user-actionable advice