Software Engineer, TPU Compiler, PhD, Early Careers
London, United KingdomPosted Aug 5, 2026
Software Engineer, TPU Compiler, PhD, Early Careers
hackajob London, England, United KingdomSoftware Engineer, TPU Compiler, PhD, Early Careers
hackajob
London, England, United Kingdom
2 weeks ago
Be among the first 25 applicants
See who hackajob has hired for this role
Use AI to assess how you fit
Get AI-powered advice on this job and more exclusive features.
hackajob is collaborating with Google to connect them with exceptional professionals for this role.Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to be versatile, display leadership qualities and be enthusiastic to take on new problems across the full-stack as we continue to push technology forward.
TPU team develops the Accelerated Linear Algebra (XLA) TPU parallelizing compiler used to partition, optimize, and run large-scale machine learning models across multiple TPU accelerators for internal (e.g. Google DeepMind) and external customers. It is a vital part of the Google Gemini software infrastructure.
Google Cloud accelerates every organization’s ability to digitally transform its business and industry. We deliver enterprise-grade solutions that leverage Google’s cutting-edge technology, and tools that help developers build more sustainably. Customers in more than 200 countries and territories turn to Google Cloud as their trusted partner to enable growth and solve their most critical business problems.
Minimum Qualifications
- Experience with coding in data structures, algorithms and software design.
- Research experience in Artificial Intelligence, Distributed Systems, Machine Learning, Data Mining, Natural Language Processing, Image Classification, Spam Fighting, or related fields.
- Work or educational experience in Machine Learning or Artificial Intelligence.
- Currently enrolled in or graduated from a PhD program.
- Experience working with parallel computing.
- Experience with compilers and compiler construction.
- Excellent debugging and programming concurrent/parallel computations, and working on accelerators such as VLIW, Vector machines, GPUs, or DSPs.
- Write product or system development code for the TPU compiler (in C++).
- Participate in, or lead design reviews with peers and stakeholders to decide amongst available technologies.
- Contribute to a compiler which scales-out machine learning models across accelerators like TPU/Graphics Processing Unit (GPU) at Google and Cloud.
- Design and implement performance optimizations and critical features, which increase the velocity of important production teams.
- Apply AI to the development of the Compiler and to the Compiler itself.
-
Seniority level
Internship -
Employment type
Full-time -
Job function
Engineering and Information Technology -
Industries
Software Development
Referrals increase your chances of interviewing at hackajob by 2x
See who you know Get notified when a new job is posted.Similar jobs
-
Senior Machine Learning Applications and Compiler Engineer, LPX
Senior Machine Learning Applications and Compiler Engineer, LPX
NVIDIA
Cambridge, England, United Kingdom 1 week ago -
Senior Compiler Engineer
Senior Compiler Engineer
NVIDIA
Cambridge, England, United Kingdom 1 week ago -
GPU Compiler Engineer - Cambridge, UK
GPU Compiler Engineer - Cambridge, UK
Qualcomm
Cambridge, England, United Kingdom 2 weeks ago -
Senior AI Software Engineer
Senior AI Software Engineer
Huawei Technologies Research & Development (UK) Ltd
Cambridge, England, United Kingdom 1 year ago -
R&D in AI Accelerator Optimization
R&D in AI Accelerator Optimization
KRAI
Cambridge, England, United Kingdom 1 week ago -
Senior Systems Research Engineer - Research Engineering - MSR Cambridge
Senior Systems Research Engineer - Research Engineering - MSR Cambridge
Microsoft
Cambridge, England, United Kingdom 18 hours ago -
Member of Technical Staff
Member of Technical Staff
Geometric
London Area, United Kingdom 3 months ago -
Software Engineer, AI/ML, PhD, Early Career
Software Engineer, AI/ML, PhD, Early Career
Google
London, England, United Kingdom 1 week ago -
Compiler Engineer
Compiler Engineer
Riverlane
Cambridge, England, United Kingdom 1 week ago -
Inference Engine Development - Member of Technical Staff
Inference Engine Development - Member of Technical Staff
Callosum
London, England, United Kingdom £101,000.00 - £192,000.00 2 weeks ago -
Training, Process Management Engineer
Training, Process Management Engineer
OpenAI
London, England, United Kingdom 1 week ago -
Member of Technical Staff - Expression of Interest
Member of Technical Staff - Expression of Interest
Callosum
London, England, United Kingdom 2 weeks ago -
Research Engineer, Pretraining Scaling - London
Research Engineer, Pretraining Scaling - London
Anthropic
Greater London, England, United Kingdom 1 week ago -
Member of Technical Staff - Mid-Training Infra
Member of Technical Staff - Mid-Training Infra
Reflection
London, England, United Kingdom 4 days ago -
Member of Technical Staff, Training Performance Engineer
Member of Technical Staff, Training Performance Engineer
Cohere
London, England, United Kingdom 1 week ago -
Senior Performance Tools Engineer
Senior Performance Tools Engineer
Arm
Cambridge, England, United Kingdom 1 week ago -
Member of Technical Staff - Research Software Engineer
Member of Technical Staff - Research Software Engineer
Reflection
London, England, United Kingdom 2 weeks ago -
Research Software Engineer
Research Software Engineer
Mistral
London, England, United Kingdom 3 weeks ago -
Member of Technical Staff, AI Infrastructure Team
Member of Technical Staff, AI Infrastructure Team
Verda
London, England, United Kingdom 1 week ago -
Research Engineer, Pretraining
Research Engineer, Pretraining
Anthropic
Greater London, England, United Kingdom 2 weeks ago -
Software Engineer, Compute Infrastructure
Software Engineer, Compute Infrastructure
OpenAI
London, England, United Kingdom $230,000.00 - $405,000.00 1 week ago -
Research Engineer, Pretraining Scaling - London
Research Engineer, Pretraining Scaling - London
SignalAI
London, England, United Kingdom 1 day ago -
Member of Technical Staff (AI Inference Engineer)
Member of Technical Staff (AI Inference Engineer)
Perplexity
London, England, United Kingdom 3 days ago -
Performance Modelling Engineer
Performance Modelling Engineer
Apple
Cambridge, England, United Kingdom 2 months ago -
Performance Analysis Engineer
Performance Analysis Engineer
Apple
London, England, United Kingdom 2 months ago -
Member of Technical Staff, Modeling
Member of Technical Staff, Modeling
Cohere
London Area, United Kingdom 1 week ago -
Performance Engineer
Performance Engineer
CommonAI CIC
Cambridge, England, United Kingdom 1 week ago
People also viewed
-
Senior Software Engineer — Runtimes and Architecture
Senior Software Engineer — Runtimes and Architecture
Cambridge, England, United Kingdom 1 week ago -
Senior Compiler Engineer
Senior Compiler Engineer
Cambridge, England, United Kingdom 1 week ago -
GPU Compiler Engineer
GPU Compiler Engineer
Cambridge, England, United Kingdom 2 weeks ago -
Compiler Engineer
Compiler Engineer
Cambridge, England, United Kingdom 1 month ago -
Research Engineer, Pretraining
Research Engineer, Pretraining
London, England, United Kingdom 1 day ago -
Developer Experience Engineer
Developer Experience Engineer
London, England, United Kingdom 2 weeks ago -
Research Engineer, Strategic Bets, DeepMind
Research Engineer, Strategic Bets, DeepMind
London, England, United Kingdom 1 week ago -
Senior Researcher - AI Computer Architecture
Senior Researcher - AI Computer Architecture
Cambridge, England, United Kingdom 2 weeks ago -
Principal AI Developer Technology Engineer
Principal AI Developer Technology Engineer
Reading, England, United Kingdom 1 week ago -
Senior CPU Compiler Performance Engineer
Senior CPU Compiler Performance Engineer
Cambridge, England, United Kingdom 2 weeks ago
Similar Searches
-
Test Engineer jobs
26,482 open jobs -
Software Engineer jobs
29,852 open jobs -
Engineer jobs
79,088 open jobs -
J2Ee Developer jobs
1,529 open jobs -
Software Engineer Intern jobs
33 open jobs -
Computer Software Engineer jobs
20,938 open jobs -
Web Development Specialist jobs
21,432 open jobs -
Software Intern jobs
98 open jobs -
Back End Developer jobs
5,847 open jobs -
Junior Software Engineer jobs
22,424 open jobs -
Graduate Software Engineer jobs
20,876 open jobs -
Game Programmer jobs
585 open jobs -
Student Software Engineer jobs
391 open jobs -
Developer Internship jobs
25 open jobs -
Developer jobs
33,470 open jobs -
Junior Developer jobs
3,039 open jobs -
Associate Software Engineer jobs
21,238 open jobs -
Frontend Developer jobs
3,363 open jobs -
Clinical Programmer jobs
170 open jobs -
Java Software Engineer jobs
3,702 open jobs -
Statistical Programmer jobs
3,220 open jobs -
Network Programmer jobs
20,656 open jobs -
Engine Programmer jobs
434 open jobs -
Client Services Engineer jobs
22,003 open jobs -
Web Developer jobs
5,705 open jobs