Senior Software Engineer, Ai/ml Infrastructure

Google Google · Big Tech · Warsaw, Poland

Senior Software Engineer focused on optimizing AI/ML infrastructure performance across the technical stack, from networking and data storage to ML models, to provide AI developers with a high-performance experience on Google's AI infrastructure. Responsibilities include designing and implementing solutions, optimizing performance, profiling, debugging, and developing tools for AI/ML infrastructure.

What you'd actually do

  1. Design and implement solutions in one or more specialized ML areas, leverage ML infrastructure, and demonstrate expertise in a chosen field.
  2. Optimize performance on Google's AI infrastructure across the technical stack.
  3. Conduct the performance profiling, debugging, and troubleshooting of AI/ML training and inference workloads.
  4. Develop tools and software for our AI/ML Infrastructure to deliver excellent developer experience.
  5. Partner closely with cross-functional, cross-regional teams to ensure our AI/ML infrastructure delivers exceptional value and drives success for our customers.

Skills

Required

  • software development
  • software design and architecture
  • ML infrastructure (e.g., model deployment, model evaluation, optimization, data processing, debugging)

Nice to have

  • distributed computing leveraging GPUs or TPUs
  • data structures and algorithms

What the JD emphasized

  • ML infrastructure
  • AI/ML training and inference workloads

Other signals

  • ML infrastructure optimization
  • AI developer experience
  • performance profiling
  • debugging AI/ML workloads