Apply on Employer Site

Google · 6 hours ago

Senior Software Engineer, TorchTPU

Sunnyvale, CA

Full-time

Onsite

Senior Level

$166K/yr - $244K/yr

5+ years exp

Google is a leading technology company, and they are seeking a Senior Software Engineer for their Core ML team. In this role, you will be responsible for the development and optimization of the PyTorch ML framework for Google Cloud's TPUs, collaborating with cross-functional teams to enhance machine learning capabilities.

AppsArtificial Intelligence (AI)Cloud StorageSearch EngineSEO

Growth Opportunities

H1B Sponsor Likely

Responsibilities

Work on AI framework development to enable PyTorch models to run on Google Cloud's TPUs and GPUs and tune for performance

Provide support for ML frameworks and compilers on Cloud TPUs and Graphics Processing Units (GPUs), enabling the training and deployment of the most advanced machine learning models, managing innovation and breakthroughs

Enable PyTorch models for generative models, computer vision (image recognition, object detection, image generation), machine translation, language modeling, rankings and recommendations, speech recognition, etc

Collaborate with other Google teams and leading researchers across the industry to continuously bring ML capabilities to our PyTorch in Cloud offering

Design, develop, test, deploy, maintain, and improve software while contributing to open-source software development

Qualification

ML designML infrastructureSoftware developmentData structuresAlgorithmsCompilersML frameworksTechnical leadershipCross-functional projectsCross-business projectsBachelor's degreeMaster's degreePhD

Required

Bachelor's degree or equivalent practical experience

5 years of experience with ML design and ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning)

5 years of experience in software development

5 years of experience testing, and launching software products, and 3 years of experience with software design and architecture

Preferred

Master's degree or PhD in Engineering, Computer Science, or a related technical field

8 years of experience with data structures and algorithms

3 years of experience in a technical leadership role leading project teams and setting technical direction

3 years of experience working in an organization involving cross-functional, or cross-business projects

Experience with compilers or ML frameworks

Benefits

Bonus

Equity

Benefits

Company

Google

Glassdoor4.4

Google specializes in internet-related services and products, including search, advertising, and software. It is a sub-organization of Alphabet.

Founded in 1998

Mountain View, California, USA

10001+ employees

https://www.google.com

H1B Sponsorship

Google has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)

Distribution of Different Job Fields Receiving Sponsorship

Represents job field similar to this job

Trends of Total Sponsorships

2025 (8763)

2024 (8872)

2023 (9682)

2022 (11626)

2021 (9109)

2020 (9785)

Funding

Current Stage

Public Company

Total Funding

$26.1M

Key Investors

Andy Bechtolsheim

2004-08-19IPO

1999-06-07Series Unknown· $25M

1998-11-01Angel· $1M

Leadership Team

Sundar Pichai

CEO

Thomas Kurian

CEO - Google Cloud

Recent News

WebProNews

Gemini’s Ad-Free Stance Challenges OpenAI’s Revenue Push

2026-01-22

PhAndroid.com

Switching from Safari to Chrome Just Got Way Easier on iPhone

2026-01-22

Android Headlines

Google App's Voice Search Finally Receives a Major Redesign on Android

2026-01-22

Company data provided by crunchbase