{bc}
Apply
Company Site

Senior Software Engineer, TorchTPU

Google
London, GBR
Senior · 5+ years experience
Onsite
Discovered 3 weeks ago
jaxpythonpytorchtensorflow

Full Job Posting

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.

Minimum 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.

Preferred qualifications:

  • 5 years of experience with data structures and algorithms.
  • Experience with ML frameworks (e.g., TensorFlow, JAX, PyTorch) and hardware performance analysis, particularly with TPUs and GPUs.
  • Experience with Compilers and Compiler construction.
  • Experience programming in Python.
  • Ability to effectively navigate and communicate in a fast-paced environment.

Apply for this job in 1 click

Skip the repetitive application forms

Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.

Sarah M.James T.Maya R.

Trusted by over 500,000 job seekers on Base Career

Start Free Today

More from this employer

More jobs at Google

Technical Program Manager III, Supply Chain Analytics, Google Cloud

Sunnyvale, USA

Discovered Today

Senior Software Engineering Manager, Infrastructure, Google Cloud Security and Privacy

Sunnyvale, USA

Discovered Today

AI Sales Specialist

Mumbai, IND

Discovered Today

Développeur de logiciels II, début de carrière

Waterloo, CAN

Discovered Today

Director, Engineering, Compute Infrastructure, Google Distributed Cloud

Sunnyvale, USA

Discovered Today

Technical Program Manager III, Network Demand Planning, Cloud Infrastructure

Sunnyvale, USA

Discovered Today

Manufacturing Test Development Engineer, Google Cloud Platforms, Infrastructure Engineering

Sunnyvale, USA

Discovered Today

Senior UX Research Program Manager, Search Ads

New York City, USA

Discovered Today