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