Research Engineer, Post-Training for Code Security Analysis

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.

In this role, you'll work with a team of elite researchers and engineers to design and implement post-training strategies that enhance Gemini’s capabilities in code security analysis. You will bring contributions to our ML innovation, post-training refinement (SFT/RLHF), advanced evaluation, and data generation to ensure our models can reliably perform safe and powerful code security analysis.

• →Design and Implement advanced post-training algorithms (SFT, RLHF, RLAIF) to optimize Gemini for code security tasks and secure coding practices.
• →Diagnose and interpret training outcomes (regressions in coding ability, gains in security reasoning), and propose solutions to improve model capabilities.
• →Actively monitor and evolve the system's performance through metric design.
• →Develop reliable automated evaluation pipelines for code security that are strongly correlated with human security expert judgment.
• →Construct complex benchmarks to probe the limits of the model’s ability to reason about control flow, memory safety, and software weakness.

We are seeking individuals who excel in fast-pacing environments and are eager to contribute to the advancement of AI. We highly value the ability to invent novel solutions to complex problems, embracing a can-do and fail-fast mindset. We are looking for someone who genuinely believes in the future of AI and is committed to devoting their energy in this field.

• BSc, MSc or PhD/DPhil degree in computer science, stats, machine learning or similar experience working in industry
• Deep understanding of statistics is strongly preferred
• Experiences in fine-tuning and adaptation of LLMs (e.g. advanced prompting,  supervised fine-tuning, RLHF)
• Strong knowledge of systems design and data structures
• Proven experience with TensorFlow, JAX, PyTorch, or similar leading deep learning frameworks
• Recent experience conducting applied research to improve the quality and training/serving efficiency of large transformer-based models
• A passion for Artificial Intelligence.
• Excellent communication skills and proven interpersonal skills, with a track record of effective collaboration with cross-functional teams

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 opportunity 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.

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

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 opportunity 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.