Google is a leading technology company that develops next-generation technologies to change how billions of users connect and interact with information. They are seeking a Software Engineer to work on the Tensor TPU Compiler, focusing on the analysis, optimization, and compilation of Machine Learning models for the Tensor TPU, enabling efficient AI models to run on devices like Pixel phones.
Work as part of the Tensor Tensor Processing Unit (TPU) compiler team, including analyzing and improving the compiler quality and performance on optimization decisions, correctness and compilation time.
Work with Tensor TPU architects to design future accelerators, the HW/SW interface, and co-optimizations of the next generation Tensor TPU architectures.
Develop parallelization and scheduling algorithms to optimize compute and data movement costs to execute ML workloads on the Tensor TPU.
Work on efficient mapping of Generative AI (GenAI) models and other key workloads into Tensor TPU instructions through the compiler.
Collaborate with ML model developers, researchers, and Tensor TPU hardware/software teams to accelerate the transition from research ideas to exceptional user experiences running on the Tensor TPU.
Qualification
Required
Bachelor’s degree or equivalent practical experience.
2 years of experience with software development in one or more programming languages (e.g. C++), or 1 year of experience with an advanced degree.
2 years of experience working with compilers (LLVM, MLIR, etc.).
Preferred
Master's degree or PhD in Computer Science or related technical fields.
2 years of experience with data structures or algorithms.
Experience with compiler development in the context of accelerator-based architectures.
Experience in optimizing ML models for inference.
Benefits
Bonus
Equity
Benefits
Google specializes in internet-related services and products, including search, advertising, and software. It is a sub-organization of Alphabet.
