Renata
Back to jobs
Google

Staff Software Engineer, AI Engines 3P TPU Inference

Google
Posted 2 weeks ago
Apply NowAsk Renata

Job Description

  • Exercise judgment to guide sustainable engineering choices for ML systems at scale.
  • Innovate next directions for infrastructure over a 12-month time horizon given a rapidly changing technology landscape.
  • Deliver impactful capability and optimization impact to Cloud and partner product areas.
  • Gracefully and incrementally migrate existing frameworks (TensorFlow, JAX, PyTorch) runtimes (TF Executor, TFRT, PJRT) and product areas custom workflows (AdBrain) onto ML Runtime, minimizing any user disruption.
  • Partner with GDM to transfer key innovations into products developed by your team and partner teams.

See Your Match Score

Sign up and Renata will show you how this job matches your skills and experience.

Get Started Free
About Google
Website
Staff Software Engineer, AI Engines 3P TPU Inference at Google | Renata