Startups · AI

Software Engineer III, ML, TPU Efficiency, YouTube

Google · United States · On-site

← All jobs
About Google

Learn more about Google. Explore our innovative AI products and services, and how we. Backed by Kleiner Perkins.

About the 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.

What they're looking for

  • Bachelor's degree or equivalent practical experience
  • 2 years of experience programming in C++ or Python
  • 2 years of experience with software design and architecture
  • 2 years of experience testing, and launching software products
  • Experience with ML model optimization
  • Experience with ML frameworks such as TensorFlow, JAX, and PyTorch, or ML compilers (e.g., accelerated linear algebra (XLA))
More about this role
  • Profile ML workloads, identify compute and memory bandwidth bottlenecks, and optimize accelerator utilization to maximize compute efficiency.
  • Explore, implement, and productionize algorithmic efficiency techniques, including low-precision quantization, knowledge distillation, parameter sharing, and attention optimizations.
  • Optimize auxiliary serving and distributed data pipelines, including data ingestion, feature transformation, embedding lookups, and memory caching to support real-time training and inference.
  • Partner closely with ML model developers and researchers to co-design hardware-friendly model architectures and deploy universal efficiency libraries.
  • Bachelor's degree or equivalent practical experience.
  • 2 years of experience programming in C++ or Python.
  • 2 years of experience with software design and architecture.
  • 2 years of experience testing, and launching software products.
  • Experience with ML model optimization.
  • Experience with ML frameworks such as TensorFlow, JAX, and PyTorch, or ML compilers (e.g., accelerated linear algebra (XLA)).
  • Master's degree or PhD in Computer Science or related technical fields.
  • Experience developing accessible...

Read the full posting on Google's site ↗

Build your edge while you search

Free tools for founders and investors, plus VC Unfiltered, our take on startups, venture and the people who build them.