Research Engineer, Gemini Post Training, Model Behavior

We are looking for a Research Engineer with proven leadership experience building large-scale machine learning systems, with expertise in modeling, data engineering and evaluations, to help us develop the best models as part of the Gemini Post-Training team, with a focus on model behavior. In this role, you will help execute the team’s strategic roadmap, identify critical modeling needs and collaborate cross-functionally and across the stack to land improvements that truly make a difference to the user.

Key responsibilities:

• Develop techniques and data pipelines to optimize frontier Gemini models on key capabilities important for model behavior
• Build robust and real-world motivated evaluations
• Partner with feature teams to prioritize modeling needs and ship gemini powered features.
• Establish best practices for code health and documentation on the team, to facilitate collaboration and reliable development.
• Mentor and guide junior engineers, fostering their professional growth and development.

• BS, MS or Ph.D. in Computer Science, Artificial Intelligence, or a related field, or equivalent practical experience
• Fluency in Python or C++ and JAX or similar ML framework
• Software engineering, ML and data engineering experience (e.g. model training / deployment, efficiency optimization, data pipeline design, ML infra)
• Experience building robust automatic or human evaluations for ML models
• Proven ability to lead engineering teams in fast-moving environments, under ambiguity
• Excellent communication and cross-functional collaboration skills, willingness to unblock others and pick up slack

• Have hands-on experience with training and debugging LLMs. Experience with  RL fine tuning methods are a plus.
• Enjoy the end-to-end process of productionizing cutting-edge research and taking projects from proof-of-concept to launch are passionate about applying AI for learning use cases obsess over the user and are excited about incorporating user feedback into the model development process
• Are well versed in Google infrastructure, including for fine-tuning and serving Gemini models

The US base salary range for this full-time position is between $166,000 - $244,000 + bonus + equity + benefits. Your recruiter can share more about the specific salary range for your targeted location during the hiring process.