Principal Engineer, Machine Learning, Vertex AI

Within Cloud AI, we focus on building highly differentiated, highly scalable, and easy-to-use machine learning products and services that enable our customers to transform their business with AI. Products such as Vertex, AutoML, TPU, and TensorFlow Enterprise comprise a portfolio that spans the needs of Data Scientists, Developers, and overall Infrastructure. We provide customers with state-of-the-art ML models and full ML model building/serving toolsets and management workflows.

As a Principal Engineer, you will be responsible for driving, deciding, and landing the technical roadmap for our Cloud AI large model platform. You will partner with Vertex platform engineering and product teams, and work with engineering managers to guide technical direction. You'll partner closely with various partner teams both within the product (e.g., Cloud DA, Cloud GKE) and cross-product partner teams (e.g., CoreML, RMI) on collaborations. In addition, you will work with customers, solutions architects, sales leads, and GTM colleagues to understand customers' use cases, and translate them into technical designs and implementations.

A significant aspect of this Principal Engineer role involves a close partnership with Google DeepMind, specifically focusing on the critical phases of post-training and large model productization within our Cloud AI offerings. You will be at the forefront of collaborating on strategies and implementations to optimize and scale DeepMind's advanced models for seamless integration into our products. This includes working on techniques for efficient deployment, inference optimization, and ensuring the reliability and performance of these large models in a production environment. Your contributions will directly impact how research translates into tangible and impactful AI solutions for our customers.