Research Engineer, Gemini: Large-Scale Pretraining + Equity

Google LLC; Mountain View (CA); On-site; USD 147,000 - 210,000; Full time

Job summary

Google DeepMind is seeking a Research Engineer for Gemini Large Scale Pretraining. The role involves embedding software engineers within research, enabling large-scale experiments, prototyping, and designing architectures to advance AI research across global teams.

Ideal candidates have a CS degree, 2 years in Python ML libraries like JAX or OSS, and 1 year ML experience, with strong collaboration and communication across time zones. Equity and benefits accompany the role.

Qualifications

• Bachelor's degree or equivalent practical experience.
• 2 years of experience with Python-based ML/scientific libraries such as JAX or OSS contributions.
• 1 year of experience with machine learning.

Responsibilities

• Work on long term projects to enable new research across multiple teams.
• Respond to time-sensitive issues in large-scale training runs and ongoing research.
• Make systematic improvements to technical workflows and tooling used across pretraining.
• Apply engineering expertise to help researchers navigate technical challenges across the codebase, model architecture, and data pipeline.
• Develop a balance between strong engineering practices and a flexible, researcher-friendly codebase.

Skills

Python ML libraries

ML experience

Education

Bachelor's degree in Computer Science or related field

Tools

JAX

Job description

Google DeepMind is seeking a Research Engineer for Gemini Large Scale Pretraining. The role involves embedding software engineers within research, enabling large-scale experiments, prototyping, and designing architectures to advance AI research across global teams.

Ideal candidates have a CS degree, 2 years in Python ML libraries like JAX or OSS, and 1 year ML experience, with strong collaboration and communication across time zones. Equity and benefits accompany the role.