Software Engineer, Machine Learning, GPU and TPU Accelerators

1 Month ago • 3 Years + • Artificial Intelligence • $136,000 PA - $200,000 PA

Job Summary

Job Description

This Software Engineer role at Google focuses on Machine Learning (ML) model performance optimization using GPUs and TPUs. Responsibilities include analyzing and improving ML model performance, contributing to performance algorithm research and development, creating automated processes for performance improvements, implementing ML solutions, and collaborating with stakeholders. The ideal candidate will have experience in software development (C++, Python), data structures/algorithms, compilers or accelerators, and ideally experience with post-training quantization and foundational models. The role requires strong collaboration skills and a commitment to efficiency and accuracy.
Must have:
  • Bachelor's degree or equivalent practical experience
  • 2 years software development (C++, Python)
  • Data structures/algorithms experience
  • 1 year experience in compilers/accelerators
  • Analyze and improve ML model performance
  • Contribute to performance algorithm research
Good to have:
  • Master's/PhD in CS or related field
  • Experience with post-training quantization
  • Experience with foundational models and LLMs
  • Understanding of ML modeling techniques
Perks:
  • Bonus
  • Equity
  • Benefits

Job Details

Minimum qualifications:

  • Bachelor's degree or equivalent practical experience.
  • 2 years of experience in software development (e.g., C++, Python), and with data structures/algorithms.
  • 1 year of experience in compilers, accelerators, or equivalent performance improvements.

Preferred qualifications:

  • Master's degree or PhD in Computer Science or a related technical field or equivalent practical experience.
  • Experience with post-training quantization, quantized aware training, or quantized training for ML models.
  • Experience with foundational models and Large Language Models.
  • Understanding of the ML modeling techniques and trends.

About the job

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full-stack as we continue to push technology forward. 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.

The US base salary range for this full-time position is $136,000-$200,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

  • Analyze and improve performance of Machine Learning (ML) models.
  • Contribute to research and development of performance algorithms.
  • Contribute to the creation of automated processes to advance the adoption of performance improvements in Google's Machine Learning (ML).
  • Implement solutions in ML areas, utilize ML infrastructure, and contribute to model enhancement and data processing.
  • Collaborate with peers and stakeholders through design and code reviews to ensure standard procedures amongst available technologies (e.g., style guidelines, checking code in, accuracy, testability, and efficiency).

Similar Jobs

Backbone - Electrical Engineer

Backbone

Atherton, California, United States (Hybrid)
6 Months ago
Google - Data Analytics Customer Engineer, Google Cloud

Google

Bengaluru, Karnataka, India (On-Site)
1 Month ago
Max Secure Software - Reverse Engineer

Max Secure Software

Pune, Maharashtra, India (Remote)
2 Months ago
Seedify - Senior UI Engineer (Unreal Engine)

Seedify

Romania (Remote)
2 Months ago
ByteDance - Research Scientist Intern (Doubao (Seed) - Machine Learning System) - 2025 Summer (PhD)

ByteDance

Seattle, Washington, 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)
2 Months ago
Zoox - Director, Process & Tools, PMO

Zoox

Foster City, California, United States (On-Site)
3 Months ago
Google - Staff Software Engineer, Platform-Aware AutoML

Google

Sunnyvale, California, United States (On-Site)
1 Month ago
Google - Software Engineer III, AI/ML

Google

Bengaluru, Karnataka, India (On-Site)
1 Month ago

Get notifed when new similar jobs are uploaded

Similar Skill Jobs

Amber - Level Designer 3D (Project Based)

Amber

Bucharest, Bucharest, Romania (Remote)
3 Months ago
Playground Games - Pipeline Technical Artist - Contract

Playground Games

Royal Leamington Spa, England, United Kingdom (Hybrid)
7 Months ago
Qualcomm - Engineer- MM Audio Test

Qualcomm

Hyderabad, Telangana, India (On-Site)
4 Months ago
Luxoft - Data & AI SME

Luxoft

Gurugram, Haryana, India (On-Site)
2 Months ago
Microsoft - Software Engineer II

Microsoft

(On-Site)
1 Month ago
Playrix - Senior C++ Software Engineer (Gameplay)

Playrix

Cyprus (Remote)
3 Months ago
Blue Yonder - Lead Software Engineer-(C++,MFC,VC++)

Blue Yonder

Bengaluru, Karnataka, India (On-Site)
4 Months ago
JPMorganChase - Associate Quantitative Research- Modelling

JPMorganChase

Mumbai, Maharashtra, India (On-Site)
3 Months ago
Luxoft - Regular BSP Developer

Luxoft

Bengaluru, Karnataka, India (Hybrid)
2 Months ago
Trend Micro - Sr. Software Engineer (XDR for Networks)

Trend Micro

Taipei City, Taiwan (On-Site)
4 Months ago

Get notifed when new similar jobs are uploaded

Jobs in Sunnyvale, California, United States

Mashgin - Deployment Engineer - North Carolina

Mashgin

Charlotte, North Carolina, United States (Remote)
3 Months ago
SSC Technologies - Account Executive

SSC Technologies

North Carolina, United States (Hybrid)
3 Months ago
Mattel  Inc  - American Girl Los Angeles  Cook Part Time

Mattel Inc

California, United States (On-Site)
1 Month ago
Meta - Software Engineer (Technical Leadership)

Meta

New York, New York, United States (On-Site)
2 Months ago
Google - Technical Writer, Foundation Services

Google

San Francisco, California, United States (On-Site)
1 Month ago
Nagarro - Associate Principal Engineer, Delivery

Nagarro

New York, New York, United States (On-Site)
3 Months ago
ION - Senior Technical Consultant - Endur

ION

Dallas, Texas, United States (On-Site)
3 Months ago
The Walt Disney Company - Massage Therapist - Part Time, $15.00/Hour + Gratuities

The Walt Disney Company

Kapolei, Hawaii, United States (On-Site)
1 Month ago
SSC Technologies - Field Sales Executive, Strategic Sales

SSC Technologies

New York, New York, United States (Hybrid)
3 Months ago
Aspire - Financial Crime Operations Lead, US

Aspire

United States (On-Site)
3 Months ago

Get notifed when new similar jobs are uploaded

Artificial Intelligence Jobs

Zoox - FPGA Development Internship/Co-op

Zoox

Foster City, California, United States (On-Site)
3 Months ago
Level AI - Backend Engineer - Customer Engineering

Level AI

Noida, Uttar Pradesh, India (Hybrid)
3 Months ago
ByteDance - Research Engineer- Foundation Model AI Platform- San Jose

ByteDance

San Jose, California, United States (On-Site)
2 Months ago
Uniphore - Software Engineer - Call Recording/Contact Center

Uniphore

Bengaluru, Karnataka, India (On-Site)
4 Months ago
Visa - Machine Learning Engineer - Sr Consultant - Cloud AI Platform

Visa

Austin, Texas, United States (Hybrid)
1 Month ago
ByteDance - AI Security Researcher - Security Flow

ByteDance

San Jose, California, United States (On-Site)
2 Months ago
Microsoft - Member of Technical Staff, AI - Multimodal

Microsoft

London, England, United Kingdom (On-Site)
1 Month ago
Enterprise Bot - Data Scientist

Enterprise Bot

Bengaluru, Karnataka, India (On-Site)
2 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.

Hyderabad, Telangana, India (On-Site)

New Taipei, New Taipei City, Taiwan (On-Site)

New York, New York, United States (On-Site)

Dublin, County Dublin, Ireland (On-Site)

Kuala Lumpur, Federal Territory Of Kuala Lumpur, Malaysia (On-Site)

Mountain View, California, United States (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