Bachelor's degree or equivalent practical experience.
2 years of software development experience in one or more programming languages.
2 years of experience in performance analysis and large-scale systems data visualization.
2 years of experience with computer architecture and performance modeling.
Preferred: Master's degree or PhD in Computer Science or related fields.
Preferred: 2 years of experience with data structures and algorithms.
Preferred: Experience in hardware-software co-design and developing accessible technologies.
Responsibilities
Track the AI market for the latest models and research breakthroughs.
Identify and validate algorithmic optimizations through targeted projects for internal customers.
Transition validated algorithms to the Frontier Model Performance and Automation teams for scaling.
Collaborate on hardware/software co-design to influence future TPU architecture.
Develop custom kernels for new algorithmic efficiency techniques and implement them in production.
Benefits
Comprehensive health and wellness programs.
Generous paid time off and holidays.
Retirement savings plans with company matching.
Opportunities for professional development and training.
Access to cutting-edge technology and resources.
Full Job Description
info_outline X In most instances, this position requires in-person interviews as part of the hiring process.
Minimum qualifications:
Bachelor's degree or equivalent practical experience.
2 years of experience with software development in one or more programming languages.
2 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging.
2 years of experience with computer architecture, performance analysis, and performance modeling.
Preferred qualifications:
Master's degree or PhD in Computer Science or related technical fields.
2 years of experience with data structures and algorithms.
Experience with hardware-software algorithmic co-design and computer architecture analysis.
Experience developing accessible technologies.
Strong background in ML models and systems.
Responsibilities
Track the AI market, including the latest open-source models, frameworks, and research breakthroughs to identify SOTA algorithmic optimization opportunities.
Identify and validate SOTA algorithmic optimizations (e.g., quantization, mxfp4) through white-glove projects for key internal customers.
Drive end-to-end impact by transitioning validated algorithms and recipes to the Frontier Model Performance (manual optimization) and Automation teams, ensuring streamlined scaling.
Collaborate on hardware/software co-design to guide future TPU architecture (e.g., nvfp4) and product capabilities.
Develop custom kernels (Pallas) for new algorithmic efficiency techniques, and put them into production for 3P/OSS and 1P models.
Individual pay is determined by factors including job-related skills, experience, and relevant education or training.
Google is a multinational technology company that specializes in Internet-related services and products. These include online advertising technologies, search engine, cloud computing, software, and hardware. Google was founded in 1998 by Larry Page and Sergey Brin while they were Ph.D. students at Stanford University. The company has grown tremendously since then and has become one of the most valuable companies in the world. Google's mission is to organize the world's information and make it universally accessible and useful.
