CPU AI Workloads and Performance Architect

2 Months ago • 5 Years + • $156,000 PA - $229,000 PA

Job Summary

Job Description

This role involves characterizing AI/ML workloads, identifying performance bottlenecks, and proposing CPU-focused optimization solutions. You will create instruction traces and benchmarks for performance analysis and microarchitecture exploration. Responsibilities include developing methodologies and tools for hardware-software co-design, driving cross-team collaboration on performance analysis infrastructure, and mentoring junior engineers. The work contributes to enhancing Google's products and the success of its CPU project, collaborating with Android and AI teams on application and benchmark performance analysis.
Must have:
  • Bachelor's degree in relevant field
  • 5+ years experience in high-performance CPU/AI architecture
  • Performance analysis and workload characterization
  • Proficiency in C/C++ and scripting languages
  • Create benchmarks and traces for performance evaluation
Good to have:
  • Master's/PhD in relevant field
  • Experience with Android/Linux performance tuning tools
  • Experience characterizing AI/ML or DSP workloads
  • Experience with system software components
Perks:
  • Bonus
  • Equity
  • Benefits

Job Details


Minimum qualifications:

  • Bachelor's degree in Electrical Engineering, Computer Engineering, Computer Science, a related field, or equivalent practical experience.
  • 5 years of experience in high-performance CPU, DSP, or AI architecture, microarchitecture.
  • Experience in performance analysis and workload characterization.
  • Experience in C/C++ and scripting languages (e.g., Python).

Preferred qualifications:

  • Master's degree or PhD in Electrical Engineering, Computer Engineering or Computer Science, with an emphasis on computer architecture.
  • Experience with Android/Linux software performance tuning tools.
  • Experience with characterizing AI/ML or DSP workloads.
  • Experience with system software components, such as Linux, drivers, and runtime.

About the job

Be part of a team that pushes boundaries, developing custom silicon solutions that power the future of Google's direct-to-consumer products. You'll contribute to the innovation behind products loved by millions worldwide. Your expertise will shape the next generation of hardware experiences, delivering unparalleled performance, efficiency, and integration.

As a CPU AI workloads and performance architect, you’ll have the opportunity to collaborate with talents in Google’s Android applications and Google’s AI teams to plan and conduct application and benchmark performance analysis. Based on the infrastructures and methodologies you build, you will explore workloads and CPU design together to provide insights for hardware and software co-design. You’ll build traces and benchmarks to provide a consistent view of the workload for all stakeholders to do performance evaluation. Your work would contribute to enhancing many Google’s products and to the success of Google’s CPU project.

Google's mission is to organize the world's information and make it universally accessible and useful. Our team combines the best of Google AI, Software, and Hardware to create radically helpful experiences. We research, design, and develop new technologies and hardware to make computing faster, seamless, and more powerful. We aim to make people's lives better through technology.

The US base salary range for this full-time position is $156,000-$229,000 + bonus + equity + benefits. Our salary ranges are determined by role, level, and location. Within the range, individual pay is determined by work location and additional factors, including job-related skills, experience, and relevant education or training. Your recruiter can share more about the specific salary range for your preferred location during the hiring process.

Please note that the compensation details listed in US role postings reflect the base salary only, and do not include bonus, equity, or benefits. Learn more about .

Responsibilities

  • Characterize AI/ML workloads and identify performance bottlenecks or patterns and propose CPU-focused AI/ML workload optimization solutions.
  • Create representative instruction traces and benchmarks for performance and microarchitecture explorations.
  • Lead the development of methodologies, tools, infrastructures to enable effective hardware-software co-design.
  • Drive cross-team collaboration to design, implement and enhance performance analysis and trace collection infrastructures.
  • Mentor and guide junior engineers.

Similar Jobs

Ion - Internship - Data Science

Ion

Pisa, Tuscany, Italy (On-Site)
8 Months ago
Epic Games - Senior Technical Product Manager

Epic Games

(On-Site)
5 Months ago
Insight Software - Senior DevOps Manager

Insight Software

Bengaluru, Karnataka, India (On-Site)
8 Months ago
bytedance - Senior Machine Learning Ops Engineer, ML System - Foundation Model

bytedance

San Jose, California, United States (On-Site)
5 Months ago
Heart Machine - Gameplay Animator

Heart Machine

Los Angeles, California, United States (Remote)
1 Month ago
Meta - Software Engineer, Machine Learning

Meta

Bellevue, Washington, United States (On-Site)
7 Months ago
Meta - Software Engineer, Machine Learning

Meta

Seattle, Washington, United States (On-Site)
7 Months ago
Google - Software Engineer III, AI/ML, Core

Google

Sunnyvale, California, United States (On-Site)
2 Months ago
Meta - Research Scientist Intern, Machine Perception for Input and Interaction (PhD)

Meta

Sausalito, California, United States (On-Site)
7 Months ago
Scale AI - Machine Learning Engineer, International Public Sector

Scale AI

United Kingdom (On-Site)
8 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Rovio Entertainment Corporation - Lead/Principal Data Engineer

Rovio Entertainment Corporation

Uusimaa, Finland (Hybrid)
3 Months ago
oportun - Senior Data Engineer

oportun

India (Remote)
1 Month ago
Rackspace Technology - Google Cloud Engineer IV

Rackspace Technology

United States (Remote)
5 Months ago
Rockstar Games - Online System Administrator

Rockstar Games

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Diligent - Staff Site Reliability Engineer

Diligent

Budapest, Hungary (Hybrid)
2 Months ago
bytedance - Software Engineer Graduate (RDMA Network- High Speed Network)

bytedance

San Jose, California, United States (On-Site)
3 Months ago
aspyr - Associate 1 Software Engineer

aspyr

Austin, Texas, United States (On-Site)
2 Months ago
Nordson Corporation - Principal Field Service Technician

Nordson Corporation

Chandler, Arizona, United States (On-Site)
1 Month ago
fluence - Platform Engineer

fluence

Houston, Texas, United States (Hybrid)
6 Months ago
Tesla - Application Support Engineer

Tesla

Brandenburg, Germany (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Austin, Texas, United States

Apple - Junior Designer, Marcom Designer

Apple

Sunnyvale, California, United States (On-Site)
1 Month ago
Everi - Field Service Technician

Everi

Buffalo, New York, United States (On-Site)
4 Months ago
Nintendo - Principal Business Systems Analyst

Nintendo

Redmond, Washington, United States (Hybrid)
11 Months ago
Nintendo - Web QA Specialist

Nintendo

Redmond, Washington, United States (Hybrid)
2 Months ago
UPF Industries  - Janitor

UPF Industries

Bridgeton, Missouri, United States (On-Site)
2 Months ago
Lytx - Software Engineer III

Lytx

San Diego, California, United States (On-Site)
2 Months ago
Apple - Machine Learning Engineering Manager (Computer Vision)

Apple

Seattle, Washington, United States (On-Site)
1 Month ago
Redhorse Corp - Senior Level Project / Program Manager

Redhorse Corp

Falls Church, Virginia, United States (On-Site)
6 Months ago
Google - Developer Operations Engineer, CPU Design Verification

Google

Austin, Texas, United States (On-Site)
2 Months ago
CyberArk - Director of Procurement

CyberArk

Newton, Massachusetts, United States (Hybrid)
1 Month ago

Get notifed when new similar jobs are uploaded

Similar Category Jobs

NVIDIA - Senior Deep Learning Engineer

NVIDIA

Redmond, Washington, United States (On-Site)
3 Months ago
Google - Photonic Engineer, Machine Learning Systems, Platforms

Google

Sunnyvale, California, United States (On-Site)
2 Months ago
bytedance - Student Researcher (Doubao (Seed) - Foundation Model - Speech & Audio) - 2025 Start (PhD)

bytedance

San Jose, California, United States (On-Site)
8 Months ago
Google - Silicon Architecture/Design Engineer

Google

Bengaluru, Karnataka, India (On-Site)
2 Months ago
Microsoft - Platform Engineering Manager

Microsoft

Redmond, Washington, United States (Hybrid)
3 Months ago
NVIDIA - Senior Solutions Architect - Generative AI

NVIDIA

Bengaluru, Karnataka, India (On-Site)
5 Months ago
bytedance - LLM Software Engineer/Researcher (Applied Machine Learning)

bytedance

Seattle, Washington, United States (On-Site)
2 Months ago
Tencent - Senior Researcher: Artificial General Intelligence (Natural Language Processing)

Tencent

Bellevue, Washington, United States (On-Site)
10 Months ago
bytedance - Student Researcher (Doubao (Seed) - Foundation Model - Vision Generative AI)

bytedance

San Jose, California, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

About The Company

Atlanta, Georgia, United States (On-Site)

New York, United States (On-Site)

London, England, United Kingdom (On-Site)

Taipei City, Taiwan (On-Site)

Kirkland, Washington, United States (On-Site)

Sunnyvale, California, United States (On-Site)

Sunnyvale, California, United States (On-Site)

Sunnyvale, California, United States (On-Site)

Kraków, Lesser Poland Voivodeship, Poland (On-Site)

View All Jobs

Get notified when new jobs are added by Google

Level Up Your Career in Game Development!

Transform Your Passion into Profession with Our Comprehensive Courses for Aspiring Game Developers.

Job Common Plug