Machine Learning Engineer, Gemini Factuality, CH

Snapshot

You are an experienced Software Engineer with a strong machine learning background who is excited about improving the quality and factual accuracy of the “Gemini” end-user product, delivering state-of-the art language models to our users.

About us

Artificial Intelligence could be one of humanity’s most useful inventions. At Google DeepMind, we’re a team of scientists, engineers, machine learning experts and more, working together to advance the state of the art in artificial intelligence. We use our technologies for widespread public benefit and scientific discovery, and collaborate with others on critical challenges, ensuring safety and ethics are the highest priority.

The role

Our team is working on improving the accuracy and factuality of responses generated by Gemini (post-trained) models, both standalone and in Retrieval Augmented Generation (RAG) settings.

We hillclimb on both the quality of RAG backends, and we build state-of-the-art language models through a variety of model training and optimization methods (Supervised Fine Tuning, Reinforcement Learning from Human Feedback, …). The team is responsible for managing data quality, model building, quality evaluation, quality monitoring, and tuning the quality of RAG backends.

We are looking for a senior software engineer with a strong ML background, working with us on making the Gemini end-user product more factual.

Key responsibilities

• Leverage model training strategies like Supervised Fine Tuning (SFT) and preference-based approaches (e.g. Reinforcement Learning from Human Feedback, Identity Preference Optimization, …) to improve the quality of language models.
• Hillclimb on the Retrieval-augmented-generation (RAG) approaches to improve the accuracy of language model responses.
• Collaborate with cross-functional and Research teams on applying state-of-the-art machine learning approaches to LLM-based products.

About you

In order to set you up for success as a machine learning-focused software engineer at Google DeepMind, we look for the following skills and experience:

• 5 years of experience with software development in one or more programming languages, and with data structures/algorithms.
• Bachelor’s degree or equivalent practical experience.
• 3 years of experience with state of the art GenAI techniques (e.g., LLMs, Multi-Modal, Large Vision Models) or with GenAI-related concepts (language modeling, computer vision).
• 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.
• 3 years of experience with ML infrastructure (e.g., model deployment, model evaluation, optimization, data processing, debugging)

In addition, the following would be an advantage: 

• PhD in Machine Learning or a related field.
• Experience with training and optimizing large language models.

At Google DeepMind, we value diversity of experience, knowledge, backgrounds and perspectives and harness these qualities to create extraordinary impact. We are committed to equal employment opportunities regardless of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, pregnancy, or related condition (including breastfeeding) or any other basis as protected by applicable law. If you have a disability or additional need that requires accommodation, please do not hesitate to let us know.