← All roles

Senior Software Engineer, TorchTPU

Google
Apply on company site
London, UKFull-timePosted 1w ago

About the role

Google Cloud software engineer role focused on AI framework development to enable PyTorch models to run on TPUs/GPUs, with emphasis on ML frameworks, compilers, and performance optimization within the PyTorch TPU team.

Responsibilities

  • Work on AI framework development to successfully enable PyTorch models to run on Google Cloud's Tensor Processing Units (TPUs) and Graphics Processing Units (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, driving innovation and breakthroughs.
  • Enable PyTorch models at massive scale for generative models, computer vision (image recognition, object detection, image generation), machine translation, language modeling, rankings and recommendations, speech recognition, etc.
  • Collaborate with other Google teams and leading researchers across the industry to continuously bring ML capabilities to our PyTorch-in-Cloud offering.
  • Contribute to open-source software development.

Qualifications

  • Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience.
  • 5 years of experience with software development in one or more programming languages (e.g., C++, C, or Python).
  • Experience in leading cross-functional software oriented projects from inception to delivery.

Skills

Required:Programming
Nice to have:Data structures and algorithmsML frameworksHardware performance analysisCompilers
Last updated Oct 23, 2026. Interested? Apply on the company site.

About Google

Google Search homepage

llmtext-to-speechvoice-ai-stack
Company profileWebsite

More open roles at Google