SIGN IN
Senior Software Engineer, Tensor Processing Units Compiler jobs in United States
cer-icon
Apply on Employer Site
company-logo

Google · 15 hours ago

Senior Software Engineer, Tensor Processing Units Compiler

Google is a leading technology company that develops next-generation technologies to enhance user connectivity and information interaction. The Senior Software Engineer will work on the MLIR/LLVM based TPU compiler, optimizing workloads and supporting new TPU hardware across multiple generations.
Artificial Intelligence (AI)Cloud ComputingAppsMarketingCloud StorageSearch EngineSEO
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Contribute to a compiler for a novel processor designed to accelerate machine learning workloads. Compile high-performance implementations of operations at a distributed scale
Work closely with users of TPUs to improve performance/efficiency and hardware designers to co-design future processors
Investigate high-level representations to effectively program large-scale, distributed, and heterogeneous systems

Qualification

CUDA C++Machine learningCompilersData structuresPerformance analysisGPU integrationSoftware designBachelor's degreeMaster's degreePhDSoft skills

Required

Bachelor's degree or equivalent practical experience
5 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture
2 years of experience working with CUDA C++ application development and 1 year of experience with Native Code, Just-In-Time (JIT), Cross, Source-to-Source or any other type of compilers
2 years of experience with machine learning algorithms and tools (e.g., TensorFlow), artificial intelligence, deep learning, or natural language processing

Preferred

Master's degree or PhD in Computer Science or related technical fields
2 years of experience with data structures and algorithms
Experience with performance, large-scale systems data analysis, visualization tools, or debugging
Experience with debugging correctness and performance issues at all levels of the stack
Experience in GPU integrating low-level CUDA work into higher-level frameworks (e.g., TF, JAX, PyTorch)
Experience with hardware/software co-design and optimizations in mid-level and low-level architecture

Benefits

Bonus
Equity
Benefits

Company

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

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
Kleiner Perkins,Sequoia CapitalAndy Bechtolsheim
2004-08-19IPO
1999-06-07Series Unknown· $25M
1998-11-01Angel· $1M

Leadership Team

leader-logo
Sundar Pichai
CEO
linkedin
leader-logo
Thomas Kurian
CEO - Google Cloud
linkedin
Company data provided by crunchbase