Google DeepMind is seeking a senior Research Engineer to lead the design and evolution of their core research ecosystem, enabling ambitious, large-scale experiments with deep learning models for scientific discovery and public benefit.
Requirements
- Expert-level proficiency in Python and JAX
- Proven experience designing, building, and pre-training LLMs and large-scale deep learning models
- Deep understanding of Transformer architectures
- Experience leading complex, multi-quarter engineering projects in a research environment
- Demonstrated ability to drive engineering best practices (e.g., design docs, code reviews) and influence technical direction
Responsibilities
- Drive the design, evolution, and long-term vision for a research codebase, ensuring its stability, simplicity, and suitability for scientific inquiry.
- Lead engineering efforts for large-scale model pre-training experiments, designing and implementing robust solutions for distributed training.
- Act as a subject matter expert on engineering best practices, championing code quality, robust design, and effective tooling.
- Partner with researchers to translate complex research ideas into practical implementations and unblock ambitious research directions.
- Identify opportunities to generalize our tools for wider adoption.
Other
- BS/MS in Computer Science or related field
