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
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
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 CompanyTotal Funding
$26.1MKey Investors
Kleiner Perkins,Sequoia CapitalAndy Bechtolsheim
2004-08-19IPO
1999-06-07Series Unknown· $25M
1998-11-01Angel· $1M
Recent News
Small Business Trends
2026-01-24
2026-01-24
Search Engine Journal
2026-01-24
Company data provided by crunchbase