Software Engineer, ML/AI Reference Models, Google Cloud

3 Months ago • 2 Years + • Research & Development • Artificial Intelligence

Job Summary

Job Description

In this role, you'll work to shape the future of AI/ML hardware acceleration. You will have an opportunity to drive cutting-edge TPU (Tensor Processing Unit) technology that powers Google's most demanding AI/ML applications. You’ll be part of a diverse team that pushes boundaries, developing custom silicon solutions that power the future of Google's TPU. You'll contribute to the innovation behind products loved by millions worldwide, and leverage your design and verification expertise to verify complex digital designs, with a specific focus on TPU architecture and its integration within AI/ML-driven systems.In this role, you will be responsible for developing functional and/or performance models for ML compute IPs and integrating them with the Cloud TPU SoC model. You will be working closely with ML and SoC architecture teams to understand the instruction set and architecture of ML IPs in good detail. You will be working closely with pre-silicon (e.g., DV, emulation), post-silicon, and Software teams who will be using these models as a part of their validation flow and aid in delivering high quality designs for next generation data center accelerators.
Must have:
  • Bachelor's degree in Computer Science or related field
  • 2 years of experience developing simulators or reference models for hardware
  • Experience with programming in C/C++ and scripting (e.g., Python/Perl)
  • Develop functional and/or performance models for ML IPs
  • Integrate the functional models with the Cloud TPU SoC model
Good to have:
  • Master's degree in Computer Science, Electrical Engineering, or a related field
  • Experience in modeling Machine Learning IPs, specifically compute IPs
  • Experience in integrating IP models with the SoC model
  • Familiarity with industry standard AI/ML topologies such as RESNET, BERT, etc.

Job Details

Minimum qualifications:

  • Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience.
  • 2 years of experience with developing simulators or reference models for hardware.
  • Experience with programming in C/C++ and scripting (e.g., Python/Perl).

Preferred qualifications:

  • Master's degree in Computer Science, Electrical Engineering, or a related field.
  • Experience in modeling Machine Learning IPs, specifically compute IPs.
  • Experience in integrating IP models with the SoC model.
  • Familiarity with industry standard AI/ML topologies such as RESNET, BERT, etc.

About the job

In this role, you’ll work to shape the future of AI/ML hardware acceleration. You will have an opportunity to drive cutting-edge TPU (Tensor Processing Unit) technology that powers Google's most demanding AI/ML applications. You’ll be part of a diverse team that pushes boundaries, developing custom silicon solutions that power the future of Google's TPU. You'll contribute to the innovation behind products loved by millions worldwide, and leverage your design and verification expertise to verify complex digital designs, with a specific focus on TPU architecture and its integration within AI/ML-driven systems.

In this role, you will be responsible for developing functional and/or performance models for ML compute IPs and integrating them with the Cloud TPU SoC model. You will be working closely with ML and SoC architecture teams to understand the instruction set and architecture of ML IPs in good detail. You will be working closely with pre-silicon (e.g., DV, emulation), post-silicon, and Software teams who will be using these models as a part of their validation flow and aid in delivering high quality designs for next generation data center accelerators.

Behind everything our users see online is the architecture built by the Technical Infrastructure team to keep it running. From developing and maintaining our data centers to building the next generation of Google platforms, we make Google's product portfolio possible. We're proud to be our engineers' engineers and love voiding warranties by taking things apart so we can rebuild them. We keep our networks up and running, ensuring our users have the best and fastest experience possible.

Responsibilities

  • Study the instruction set and architecture of the ML IPs and drive discussions on the delta features from the previous generation.
  • Develop functional and/or performance models for the ML IPs. Integrate the functional models with the Cloud TPU SoC model and deliver the single source of truth architectural reference.
  • Work with pre-silicon verification and post-silicon validation teams to deploy the models into their validation flows.
  • Work with compiler and software teams to enable them to left-shift their development activity.

Similar Jobs

Meta - Network Production Engineer, Network Infrastructure

Meta

Boston, Massachusetts, United States (On-Site)
3 Months ago
ByteDance - Software Engineer Graduate (Multi Cloud CDN) - 2025 Start (BS/MS)

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Dolby Laboratories - Sr. Generative Computer Vision Research

Dolby Laboratories

Bengaluru, Karnataka, India (Hybrid)
4 Months ago
Keywords Studios (Player Support) - Tools Engineer

Keywords Studios (Player Support)

Beijing, Beijing, China (On-Site)
3 Months ago
Fabric - Principal Design Verification Engineer

Fabric

Austin, Texas, United States (On-Site)
4 Months ago
Fabric - Applied Cryptographer, ZKP Research

Fabric

France (Remote)
4 Months ago
Codvoai - Senior Data Scientist

Codvoai

Pune, Maharashtra, India (Remote)
1 Year ago
Riot Games - Senior Game Product Manager, Social Systems - 2XKO

Riot Games

Los Angeles, California, United States (On-Site)
4 Months ago
Google - Software Engineering Manager II, Google Kubernetes Engine

Google

Kraków, Lesser Poland Voivodeship, Poland (On-Site)
3 Months ago
Valeo - Deputy Tech Lead

Valeo

Chennai, Tamil Nadu, India (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Luxoft - Android Developer / CTS Expert

Luxoft

New York, United States (Remote)
3 Months ago
Larian Studios - Internship - Graphics Programmer

Larian Studios

Barcelona, Catalonia, Spain (On-Site)
4 Months ago
Blizzard Entertainment - Senior Software Engineer, Engine Systems

Blizzard Entertainment

Irvine, California, United States (Hybrid)
3 Months ago
Matic Robots - Research Engineer

Matic Robots

Mountain View, California, United States (On-Site)
4 Months ago
ByteDance - Senior Software Engineer, Global Payment Risk & Compliance

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Google - Software Engineer III, Infrastructure, Google Cloud Storage

Google

Seattle, Washington, United States (On-Site)
3 Months ago
Ubisoft - Technical Artist

Ubisoft

Taguig, Metro Manila, Philippines (On-Site)
9 Months ago
HP - Engineering Internship - Imaging and Print

HP

Corvallis, Oregon, United States (On-Site)
4 Months ago
Meta - Embedded Software Engineer, Firmware - Reality Labs

Meta

Austin, Texas, United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Bengaluru, Karnataka, India

Vigaet - Internship - Mechanical Engineer

Vigaet

Bengaluru, Karnataka, India (On-Site)
3 Months ago
Beghou Consulting - Staffing Specialist

Beghou Consulting

Pune, Maharashtra, India (Hybrid)
5 Months ago
PwC - IN_Senior Associate_Quality Assurance Consultant_Advisory Corporate_Bangalore

PwC

Bengaluru, Karnataka, India (On-Site)
4 Months ago
Maersk Careers - Senior Analyst, ATR

Maersk Careers

Chennai, Tamil Nadu, India (On-Site)
4 Months ago
Upstox - SDE - I (IOS)

Upstox

Bengaluru, Karnataka, India (On-Site)
4 Months ago
CloudHire - Senior Python Developer

CloudHire

Bengaluru, Karnataka, India (Remote)
4 Months ago
Sumo Logic - Senior Application Security Engineer-I

Sumo Logic

Pune, Maharashtra, India (Remote)
3 Months ago
Aryaka - Member of Technical Staff - QA Security

Aryaka

Bengaluru, Karnataka, India (On-Site)
5 Months ago
Rivos - Silicon DFT - Full time

Rivos

Bengaluru, Karnataka, India (Hybrid)
4 Months ago
Workato - Technical Architect

Workato

Hyderabad, Telangana, India (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Research & Development Jobs

QASCOM - R&D Navigation Engineer

QASCOM

Italy (On-Site)
8 Months ago
Mulya Technologies - SENIOR LOGIC DESIGN ENGINEER – SMP Interrupt Controller

Mulya Technologies

Bengaluru, Karnataka, India (On-Site)
4 Months ago
ByteDance - Tech Lead Manager - Code AI/GPT

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Google - Camera 3A/ISP Engineer, Pixel Camera

Google

(On-Site)
3 Months ago
ByteDance - Software Engineer Intern (Doubao (Seed) - Machine Learning System) - 2025 Summer (PhD)

ByteDance

San Jose, California, United States (On-Site)
3 Months ago
Krafton  - [Publishing Platform Div.] Unity Platform SDK Developer (3년 이상)

Krafton

Seoul, South Korea (On-Site)
3 Months ago
Zippin - Senior Embedded Software Engineer

Zippin

Bengaluru, Karnataka, India (Hybrid)
5 Months ago
Nielsen Holdings - Manager Software Engineer-AM-TECH-DA-06

Nielsen Holdings

Bengaluru, Karnataka, India (On-Site)
4 Months ago
Match Group - Senior ML Software Engineering Team Leader

Match Group

Seoul, South Korea (Hybrid)
4 Months ago
MIPS - Embedded Software Engineer – RTOS – CPU/Platform Software Team

MIPS

Pune, Maharashtra, India (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

About The Company

A problem isn't truly solved until it's solved for all. Googlers build products that help create opportunities for everyone, whether down the street or across the globe. Bring your insight, imagination and a healthy disregard for the impossible. Bring everything that makes you unique. Together, we can build for everyone.

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