Machine Learning Engineer, Gemini Safety
Zurich, Switzerland
DeepMind
Artificial intelligence could be one of humanity’s most useful inventions. We research and build safe artificial intelligence systems. We're committed to solving intelligence, to advance science...Snapshot
You are an experienced Machine Learning Engineer with a strong engineering and research background who is excited about improving safety and alignment 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 works on developing safe, aligned, and helpful Gemini models across all modalities (text, images, audio) as well as novel ways of evaluating them through superhuman autoraters or automated red teaming.
To that end, we are advancing the state-of-the-art on the entire post-training pipeline. This includes innovations in RLHF/RL*F tailored to safety and alignment, novel and scalable ways of evaluating models to find weaknesses, as well as improvements in model understanding to guide future research.
We are looking for a Senior Machine Learning Engineer excited to push the state of the art in these areas.
Key responsibilities
- Apply and innovate on state-of-the-art modeling techniques with the goal of reducing safety violations while increasing human alignment using post-training approaches (e.g. SFT, RLHF/RL*F).
- Collaborate with peers and cross-functional research teams on applying state-of-the-art machine learning approaches to LLM-based products.
- Propose and quickly prototype novel ideas and implement advances in Gemini's production systems.
- Have bias for action: our team operates in a high-reward and fast-paced environment.
- In summary: we are looking for a Machine Learning Engineer with strong intuition and the ability to drive ambiguous projects end-to-end to push the boundaries of Safety and Alignment on LLMs.
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:
- MSc degree in computer science or machine learning or similar experience working in industry.
- 3+ years of experience with ML modeling and generative AI, working with data, model evaluations, and quality iterations.
- Proven knowledge and experience of Python, one or several machine learning frameworks (Jax, TensorFlow, or PyTorch), as well as algorithms and data structures.
- Ability to grow in high-stakes, high-visibility environments.
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.
Application deadline: 12pm GMT Friday April 16th 2025
Note: In the event your application is successful and an offer of employment is made to you, any offer of employment will be conditional on the results of a background check, performed by a third party acting on our behalf. For more information on how we handle your data, please see our Applicant and Candidate Privacy Policy.
* Salary range is an estimate based on our AI, ML, Data Science Salary Index 💰
Tags: Computer Science Engineering Gemini Generative AI JAX LLMs Machine Learning PhD Privacy Python PyTorch Research RLHF TensorFlow
More jobs like this
Explore more career opportunities
Find even more open roles below ordered by popularity of job title or skills/products/technologies used.