Research Engineer, Gemini Large Scale Pretraining

$147K - $210K Mountain View, CA, US Mid Level Research Engineer

Interested in this Research Engineer role at DeepMind?

Apply Now →

Skills & Technologies

GeminiJaxPython

About This Role

AI job market dashboard showing open roles by category

Note: By applying to this position you will have an opportunity to share your preferred working location from the following: London, UK; Mountain View, CA, USA.

Minimum qualifications:

  • Bachelor's degree in Computer Science, or a related technical field, or equivalent practical experience.
  • 2 years of experience with software engineering involving Python-based ML/scientific libraries such as JAX or internship experience in these areas or Open Source Software (OSS) contributions.
  • 1 year of experience with machine learning.

Preferred qualifications:

  • Experience with large-scale system design, distributed systems, or performance engineering.
  • A good understanding of engineering best practices, codebase design, and testing.
  • Familiarity with how modern large language models are trained.
  • Ability to collaborate with, and communicate effectively with, people with different skill sets and needs, and who may be distributed across different time zones.

About the job

At Google, research-focused Software Engineers are embedded throughout the company, allowing them to setup large-scale tests and deploy promising ideas quickly and broadly. Ideas may come from internal projects as well as from collaborations with research programs at partner universities and technical institutes all over the world.

From creating experiments and prototyping implementations to designing new architectures, engineers work on real-world problems including artificial intelligence, data mining, natural language processing, hardware and software performance analysis, improving compilers for mobile platforms, as well as core search and much more. But you stay connected to your research roots as an active contributor to the wider research community by partnering with universities and publishing papers.

Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high-quality product innovation for billions of users. We use our technologies for widespread public benefit and scientific discovery, ensuring safety and ethics are always our highest priority.

We are pushing the boundaries across multiple domains. Our global teams offer diverse learning opportunities and varied career pathways for those driven to achieve exceptional results through collective effort. Individual pay is determined by factors including job-related skills, experience, and relevant education or training.

US: $147000 - $210000 (USD) + 15% bonus target + equity + benefits

Learn more about benefits at Google.Responsibilities

  • Work on long term projects to enable new research, which involve multiple team members and require good software engineering and design.
  • Respond to time-sensitive issues arising in large-scale training runs, and in ongoing research.
  • Make systematic improvements to technical workflows and tooling used across pretraining.
  • Apply engineering expertise to help researchers navigate technical challenges. This requires both a deep understanding of the codebase and a broad understanding of the tech stack, the model architecture and data pipeline.
  • Develop the right balance between good engineering practices and having a flexible codebase which researchers find easy to use and extend.

Google is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. See also Google's EEO Policy and EEO is the Law. If you have a disability or special need that requires accommodation, please let us know by completing our Accommodations for Applicants form.

Salary Context

This $147K-$210K range is below the median for Research Engineer roles in our dataset (median: $185K across 72 roles with salary data).

View full Research Engineer salary data →

Role Details

Company DeepMind
Title Research Engineer, Gemini Large Scale Pretraining
Location Mountain View, CA, US
Experience Mid Level
Salary $147K - $210K
Remote No

About This Role

Research Engineers bridge the gap between research and production. They implement papers, build experiment infrastructure, optimize training pipelines, and make research prototypes production-ready. They're the engineers who make research work at scale.

The role sits at a unique intersection. You need to understand the math well enough to implement novel architectures correctly, and you need the engineering chops to make them run efficiently on distributed systems. When a research scientist has a breakthrough idea, you're the person who turns it from a notebook prototype into a training pipeline that runs on 256 GPUs.

Across the 11,520 AI roles we're tracking, Research Engineer positions make up 2% of the market. At DeepMind, this role fits into their broader AI and engineering organization.

Research Engineer roles are growing as AI labs recognize that research velocity depends on engineering quality. The role is less competitive than Research Scientist (no PhD required), but the bar for engineering skill is very high. These roles are concentrated at major labs and well-funded startups.

What the Work Looks Like

A typical week involves: implementing a new attention mechanism from a recent paper, profiling and optimizing a training pipeline that's bottlenecked on data loading, building evaluation infrastructure for a new benchmark, debugging distributed training issues across a GPU cluster, and pair-programming with a research scientist on their latest experiment. The work is deeply technical.

Research Engineer roles are growing as AI labs recognize that research velocity depends on engineering quality. The role is less competitive than Research Scientist (no PhD required), but the bar for engineering skill is very high. These roles are concentrated at major labs and well-funded startups.

Skills Required

Gemini (2% of roles) Jax (1% of roles) Python (20% of roles)

Strong software engineering fundamentals plus ML knowledge. Python, C++, and CUDA experience are common requirements. You'll need to read papers and turn ideas into working code. Distributed systems experience (especially distributed training) is highly valued. Performance optimization skills separate great candidates from good ones.

Experience with large-scale training infrastructure (FSDP, DeepSpeed, Megatron), GPU programming (CUDA, Triton), and the internals of ML frameworks (PyTorch internals, custom autograd functions) is what makes candidates stand out. The best research engineers can debug issues that span the full stack from GPU memory management to numerical precision to algorithmic correctness.

Strong postings mention the team's recent research, the infrastructure scale, and the specific technical challenges. They often list the research areas you'd support. Look for roles that emphasize both implementation quality and research understanding.

Compensation Benchmarks

Research Engineer roles pay a median of $252,000 based on 476 positions with disclosed compensation. Mid-level AI roles across all categories have a median of $193,200. This role's midpoint ($178K) sits 29% below the category median. Disclosed range: $147K to $210K.

Across all AI roles, the market median is $213,480. Top-quartile compensation starts at $264,000. The 90th percentile reaches $318,400. For comparison, the highest-paying categories include AI Safety ($266,050) and AI Engineering Manager ($253,461). By seniority level: Entry: $112,320; Mid: $193,200; Senior: $230,000; Director: $272,250; VP: $235,830.

DeepMind AI Hiring

DeepMind has 14 open AI roles right now. They're hiring across Research Engineer, AI/ML Engineer, AI Product Manager, Research Scientist. Positions span Mountain View, CA, US, Washington, DC, US, New York, NY, US. Compensation range: $210K - $364K.

Location Context

Across all AI roles, 6% (708 positions) offer remote work, while 10,737 require on-site attendance. Top AI hiring metros: New York (3,257 roles, $221,000 median); San Francisco (2,766 roles, $260,550 median); Los Angeles (1,580 roles, $210,000 median).

Career Path

Common paths into Research Engineer roles include Software Engineer, ML Engineer, Research Intern.

From here, career progression typically leads toward Senior Research Engineer, Research Scientist, ML Architect.

This is one of the best entry points into AI research without a PhD. Build a strong engineering portfolio with ML projects, contribute to open-source ML frameworks, and demonstrate that you can implement complex ideas correctly and efficiently. The transition to Research Scientist is possible with published first-author work, which some research engineer roles support.

What to Expect in Interviews

Technical screens test both engineering skill and research understanding. Expect coding rounds with performance-critical implementations (GPU optimization, efficient data loading). Be prepared to discuss papers relevant to the team's research area and explain how you'd implement key ideas. System design questions focus on training infrastructure: distributed training, experiment tracking, and compute resource management.

When evaluating opportunities: Strong postings mention the team's recent research, the infrastructure scale, and the specific technical challenges. They often list the research areas you'd support. Look for roles that emphasize both implementation quality and research understanding.

AI Hiring Overview

The AI job market has 11,520 open positions tracked in our dataset. By seniority: 140 entry-level, 5,630 mid-level, 4,176 senior, and 1,574 leadership roles (Director, VP, C-Level). Remote roles make up 6% of the market (708 positions). The remaining 10,737 roles require on-site or hybrid attendance.

The market median for AI roles is $213,480. Top-quartile compensation starts at $264,000. The 90th percentile reaches $318,400. Highest-paying categories: AI Safety ($266,050 median, 70 roles); AI Engineering Manager ($253,461 median, 58 roles); Research Engineer ($252,000 median, 476 roles).

Research Engineer roles are growing as AI labs recognize that research velocity depends on engineering quality. The role is less competitive than Research Scientist (no PhD required), but the bar for engineering skill is very high. These roles are concentrated at major labs and well-funded startups.

The AI Job Market Today

The AI job market spans 11,520 open positions across 16 role categories. The largest categories by volume: AI/ML Engineer (8,769), AI Software Engineer (586), AI Product Manager (511). These three account for the majority of open positions, though smaller categories often have higher per-role compensation because of specialized skill requirements.

The seniority mix tells a story about where AI teams are in their maturity. Entry-level roles (140) are outnumbered by mid-level (5,630) and senior (4,176) positions, reflecting that most companies are past the 'build a team from scratch' phase and need experienced engineers who can ship production systems. Leadership roles (Director, VP, C-Level) total 1,574 positions, representing the bottleneck between technical execution and organizational strategy.

Remote work availability sits at 6% of all AI roles (708 positions), with 10,737 requiring on-site or hybrid attendance. The remote share has stabilized after the post-pandemic correction. Senior and specialized roles (Research Scientist, ML Architect) are more likely to be remote-eligible than entry-level positions, partly because experienced hires have more negotiating power and partly because these roles require less hands-on mentorship.

AI compensation is structured in clear tiers. The market median sits at $213,480. Top-quartile roles start at $264,000, and the 90th percentile reaches $318,400. These figures include base salary with disclosed compensation. Total compensation (including equity, bonuses, and sign-on) runs 20-40% higher at companies that offer those components.

Category matters for compensation. AI Safety roles lead at $266,050 median, while Prompt Engineer roles sit at $140,000. The spread between highest and lowest-paying categories reflects the premium on specialized technical skills versus broader analytical roles.

The most in-demand skills across all AI postings: Python (2,267 postings), Aws (1,178 postings), Azure (972 postings), Rag (924 postings), Gcp (715 postings), Pytorch (586 postings), Claude (572 postings), Kubernetes (559 postings). Python dominates, appearing in the vast majority of role descriptions regardless of category. Cloud platform experience (AWS, GCP, Azure) is the second most common requirement. The newer entrants to the top skills list (RAG, vector databases, LLM APIs) reflect the shift from traditional ML toward generative AI applications.

Frequently Asked Questions

Based on 476 roles with disclosed compensation, the median salary for Research Engineer positions is $252,000. Actual compensation varies by seniority, location, and company stage.
Strong software engineering fundamentals plus ML knowledge. Python, C++, and CUDA experience are common requirements. You'll need to read papers and turn ideas into working code. Distributed systems experience (especially distributed training) is highly valued. Performance optimization skills separate great candidates from good ones.
About 6% of the 11,520 AI roles we track offer remote work. Remote availability varies by company and seniority level, with senior and leadership roles more likely to offer location flexibility.
DeepMind is among the companies actively hiring for AI and ML talent. Check our company profiles for detailed breakdowns of open roles, salary ranges, and hiring trends.
Common next steps from Research Engineer positions include Senior Research Engineer, Research Scientist, ML Architect. Progression depends on whether you lean toward technical depth, people management, or product strategy.

Get Weekly AI Career Intelligence

Salary data, skills demand, and market signals from 16,000+ AI job postings. Every Monday.