AI/ML Infrastructure Software Engineer

About the Job

The AI Flow Platform and suite of products are used by thousands of Googlers across all product areas to drive business impact for Google in innumerable ways (e.g., finding insights in data, building classical ML models, experimenting with Gemini, and more). Our team provides a dependable, compliant, and scalable platform that enables product teams to iterate rapidly on new ideas and to take the best ideas for production.

Responsibilities

• Develop and maintain our production service, written in C++ and Python, and further the platform goal of incorporating prompt optimization, fine tuning, hyperparameter optimization, and ML best practices into Google product development.
• Design and implement customer requested ML and AI features end-to-end from requirements-gathering and design-authoring, through implementation and landing impact in production.
• Explore charter expansion based on next-generation product requirements from internal and external customers.

Qualifications

Minimum qualifications:

• Bachelor's degree or equivalent practical experience.
• 8 years of experience with software development in one or more programming languages (e.g., Python, C, C++, Java, JavaScript).
• Experience in ML Infrastructure Development or ML Performance Engineering.

Preferred qualifications:

• Experience with Machine Learning (ML) development process.
• Expertise in C++ and Python, Responsible ML, ML Explainability, ML Fairness Tensorflow , JAX, Keras, Gemax TFX, TFlex, Xflow or ML Performance Engineering.
• Knowledge of Google production infrastructure.
• Ability to collaborate with partners on applying ML to their use cases.
• Passion for building infrastructure to increase the velocity of ML development.