Google · 1 day ago
Senior Staff Software Engineer, Torch TPU
Sunnyvale, CA
Full-time
Onsite
Senior Level, Lead/Staff
$248K/yr - $349K/yr
8+ years expGoogle's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information. The Senior Staff Software Engineer will work on AI framework development, providing support for ML frameworks and compilers on Cloud TPUs and GPUs, and collaborating with teams to enhance ML capabilities.
AppsArtificial Intelligence (AI)Cloud StorageSearch EngineSEO
Responsibilities
Work on AI framework development to successfully enable PyTorch models to run on Google Cloud's TPUs and GPUs and tune for peak performance
Provide comprehensive support for ML frameworks and compilers on Cloud TPUs and GPUs, enabling the training and deployment of the most advanced machine learning models and driving innovation and breakthroughs
Enable PyTorch models at massive scale for generative models, computer vision, machine translation, language modeling, rankings and recommendations, speech recognition, etc
Collaborate with partner 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
Required
Bachelor's degree or equivalent practical experience
8 years of experience in software development
7 years of experience leading technical project strategy, ML design, and working with ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning)
5 years of experience with one or more of the following: Speech/audio (e.g., technology duplicating and responding to the human voice), reinforcement learning (e.g., sequential decision making), ML infrastructure, or specialization in another ML field
5 years of experience with design and architecture; and testing/launching software products
Experience in Python and C++ programming
Preferred
Master's degree or PhD in Engineering, Computer Science, or a related technical field
8 years of experience with data structures and algorithms
5 years of experience in a technical leadership role leading project teams and setting technical direction
3 years of experience working in a complex, matrixed organization involving cross-functional, or cross-business projects
Experience in performance analysis and debugging, including for systems that span multiple interconnected hosts
Experience in contributing to PyTorch or JAX
Company
Google specializes in internet-related services and products, including search, advertising, and software. It is a sub-organization of Alphabet.
10001+ employees
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 CompanyTotal Funding
$26.1MKey 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
2026-01-22
2026-01-22
2026-01-22
Company data provided by crunchbase