CPU AI Workloads and Performance Architect

2 Months ago • 5 Years + • $150,000 PA - $223,000 PA

Job Summary

Job Description

As a CPU AI Workloads and Performance Architect at Google, you'll collaborate with Android and AI teams to analyze application and benchmark performance. You'll build infrastructures and methodologies for workload exploration and CPU design, creating traces and benchmarks for performance evaluation. Responsibilities include characterizing AI/ML workloads, identifying performance bottlenecks, proposing optimization solutions, developing methodologies and tools for hardware-software co-design, and mentoring junior engineers. This role contributes to enhancing Google products and the success of Google's CPU project. You will work on developing custom silicon solutions that power the future of Google's direct-to-consumer products.
Must have:
  • Bachelor's degree in relevant field
  • 5+ years experience in high-performance CPU/AI architecture
  • Performance analysis & workload characterization
  • C/C++ and scripting (Python)
  • AI/ML workload optimization
Good to have:
  • Master's/PhD in relevant field
  • Android/Linux performance tuning
  • Experience with system software components (Linux, drivers, runtime)
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 $150,000-$223,000 + bonus + equity + benefits. Our salary ranges are determined by role, level, and location. The range displayed on each job posting reflects the minimum and maximum target salaries for the position across all US locations. 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

GoMotive - Senior Data Scientist, Finance

GoMotive

United States (Remote)
4 Months ago
ByteDance - AI Product Manager, User Operation/Growth

ByteDance

(On-Site)
2 Months ago
ByteDance - Student Researcher (Doubao (Seed) - Foundation Model - Vision Generative AI)

ByteDance

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

Google

London, England, United Kingdom (On-Site)
2 Months ago
N-iX - Technical Lead Data Engineer

N-iX

Ukraine (Hybrid)
4 Months ago
N-iX - Senior Computer Vision Engineer

N-iX

Argentina (Remote)
2 Months ago
QUANTIC DREAM - Programmeur Intelligence Artificielle

QUANTIC DREAM

Montreal, Quebec, Canada (Hybrid)
8 Months ago
Discord - Director of Machine Learning, Safety

Discord

San Francisco, California, United States (Remote)
5 Months ago
Google - Staff Software Engineer, Platforms

Google

Sunnyvale, California, United States (On-Site)
2 Months ago
NVIDIA - Solutions Architect, AI and ML

NVIDIA

Redmond, Washington, United States (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

N-iX - Senior Engineer with AWS Greengrass Expertise

N-iX

Ukraine (Hybrid)
2 Months ago
Google - Software Engineer, AICore, Knowledge and Information

Google

Taipei City, Taiwan (On-Site)
2 Months ago
Riot Games - Principal Insights Analyst - Player Platform

Riot Games

Los Angeles, California, United States (On-Site)
7 Months ago
Maersk Careers - Elixir Software Engineer - Energy Transition

Maersk Careers

Porto, Porto District, Portugal (Remote)
3 Months ago
Google - Product Engineer, Machine Learning Accelerators

Google

Sunnyvale, California, United States (On-Site)
2 Months ago
GoReel - Senior DevOps

GoReel

Bratislava, Bratislava Region, Slovakia (Hybrid)
2 Months ago
Tencent - WeChat Product Management Intern

Tencent

(On-Site)
2 Months ago
MindArk - Game Engineer (AI Team)

MindArk

Gothenburg, Västra Götaland County, Sweden (Hybrid)
8 Months ago
NVIDIA - Malware Research Architect

NVIDIA

Germany (Remote)
3 Months ago
Rockstar Games - Animation R&D Programmer: Retargeting

Rockstar Games

Edinburgh, Scotland, United Kingdom (On-Site)
2 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Austin, Texas, United States

Google - Business Systems Analyst, Supply Chain Logistics

Google

Sunnyvale, California, United States (On-Site)
2 Months ago
NVIDIA - Senior System Software Engineer - Dynamo and Triton Inference Server

NVIDIA

California, United States (Remote)
2 Months ago
Google - CPU Power System Modeling Engineer

Google

Austin, Texas, United States (On-Site)
2 Months ago
Thatgamecompany - General Art

Thatgamecompany

United States (Remote)
2 Months ago
Netflix - Manager of Creative, Experiences

Netflix

Los Angeles, California, United States (On-Site)
4 Months ago
Netflix - ML Software Engineer (L4/L5) - Media Algorithms

Netflix

Los Angeles, California, United States (On-Site)
2 Months ago
Wolters Kluwer - Associate Director, Technology Sales Support - CPM Pre-Sales

Wolters Kluwer

New York, New York, United States (Remote)
7 Months ago
Google - Staff Software Engineer, Host Networking Functions

Google

Sunnyvale, California, United States (On-Site)
2 Months ago
Netflix - Product Design Manager

Netflix

Los Gatos, California, United States (On-Site)
2 Months ago
Trek - Seasonal Sales Associate

Trek

Highlands Ranch, Colorado, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Similar Category Jobs

Airlab Inc  - Junior Programmer Artificial Intelligence

Airlab Inc

Quebec, Canada (On-Site)
2 Months ago
Google - Software Engineer III, AI/ML, Google Cloud Platforms

Google

Sunnyvale, California, United States (On-Site)
2 Months ago
NVIDIA - Senior System Software Engineer - Dynamo and Triton Inference Server

NVIDIA

California, United States (Remote)
2 Months ago
Google - Customer Engineer, Applied AI

Google

Munich, Bavaria, Germany (On-Site)
2 Months ago
Canva - Senior Computer Vision Engineer - Photo AI

Canva

Prague, Czechia (Remote)
4 Months ago
Google - Customer Engineer, Cloud AI, Google Cloud

Google

New York, New York, United States (On-Site)
2 Months ago
Google - AI/ML Engineer, National Security, Public Sector

Google

Reston, Virginia, United States (On-Site)
2 Months ago
Meta - Software Engineer, Machine Learning

Meta

Pittsburgh, Pennsylvania, United States (On-Site)
7 Months ago
Rackspace Technology - AI/ML Architect

Rackspace Technology

Vietnam (Remote)
3 Months ago
Google - Software Engineer, XProf

Google

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

Get notifed when new similar jobs are uploaded

About The Company

San Diego, California, United States (On-Site)

Los Angeles, California, United States (On-Site)

Massachusetts, United States (Remote)

United States (On-Site)

Bengaluru, Karnataka, India (On-Site)

Chicago, Illinois, United States (Hybrid)

Bengaluru, Karnataka, India (On-Site)

Regensburg, Bavaria, Germany (Remote)

Lanham, Maryland, United States (On-Site)

Toronto, Ontario, Canada (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