Research Engineer, Post-Training for Code Security Analysis

DeepMind • Mountain View, CA

6d • $141,000 - $291,000

About The Position

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

Requirements

• 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

Responsibilities

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

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level