Software Engineer Iii, ML Infrastructure, Google Cloud

Google Google · Big Tech · Pune, Maharashtra, India

Software Engineer III, ML Infrastructure for Google Cloud's Vertex Training platform, focusing on building components for large-scale distributed training, pre-training, and post-training software to enable customer ML workloads. Collaborates with partner teams and customers to identify pain points and drive product excellence.

What you'd actually do

  1. Build components to enable large-scale distributed training with focus on usability, performance and resiliency.
  2. Collaborate with partner teams, dependencies (e.g., Google Compute Engine (GCE), Google Kubernetes Engine (GKE), Cloud Tensor Processing Unit (Cloud TPU), Core Infra, CoreML) to develop pre-training/post-training software to enable customer run reliable and performant workloads.
  3. Work with Product Team, Account teams and customers to identify key pain points, define scope of the problems, translate them into projects and lead them to execution.
  4. Contribute to product excellence, and improve overall product quality and user experience.

Skills

Required

  • software development
  • Python
  • C++
  • Java
  • machine learning infrastructure
  • distributed machine learning
  • distributed computing
  • Cloud APIs

Nice to have

  • large-scale distributed machine learning
  • ML performance optimizations
  • PyTorch
  • Cloud and ML products
  • back-end system development
  • Google Infrastructure
  • OSS AI/ML Job scheduling software

What the JD emphasized

  • machine learning infrastructure
  • distributed machine learning
  • large-scale distributed training

Other signals

  • ML infrastructure
  • distributed machine learning
  • large-scale distributed training