Staff Software Engineer, AI Engines 3P TPU Inference - Google

Minimum qualifications:

• Bachelor’s degree or equivalent practical experience.
• 8 years of experience in software development.
• 5 years of experience testing, and launching software products, and 3 years of experience with software design and architecture.
• 5 years of experience with one or more of the following: Speech/audio (e.g., technology duplicating and responding to the human voice), reinforcement learning (e.g., sequential decision making), ML infrastructure, or specialization in another ML field.
• 5 years of experience with ML design and ML infrastructure (e.g., model deployment, model evaluation, data processing, debugging, fine tuning).

Preferred qualifications:

• Master’s degree or PhD in Engineering, Computer Science, or a related technical field.
• 8 years of experience with data structures and algorithms.
• 3 years of experience in a technical leadership role leading project teams and setting technical direction.
• 3 years of experience working in a complex, matrixed organization involving cross-functional, or cross-business projects.
• Experience with TPUs, TPU system design, and GPUs.
• Expertise in ML compilers and runtimes.

Responsibilities

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