Browse Jobs
Explore open roles across India.Explore active software roles. Sign up to apply and get AI match scores.
Top MatchesTop Job Matches
Loading jobs…
Back to Browse Jobs
Google DeepMind
Research Engineer, Pretraining, DeepMind
Google DeepMind
New York•On-site•Salary•5-8 Yrs•1 day ago
Job description
Research EngineerPre-trainingLanguage Model ArchitecturesAI Research
Overview
Google DeepMind is seeking a Research Engineer focused on inference-optimized model design for Gemini pretraining. Engineers work on real-world problems including natural language processing, machine learning architectures, and product innovation for millions of users.
Responsibilities
- Focus on inference-optimized model design for Gemini pretraining
- Understand common Accelerated Linear Algebra (XLA) primitives and how JAX code runs on TPUs in practice
- Work across the Large Language Model (LLM) preparation stack (e.g., pretraining, fine tuning, serving) to deliver strong models
Qualifications
- Bachelor's degree in Computer Science, Mathematics, Applied Stats, Machine Learning, a related field, or equivalent practical experience
- 5 years of experience with inference-optimized model design
- Experience working with product teams
Nice to have
- Experience conducting applied research to improve the quality and training/serving efficiency of large transformer-based models
- Experience with fine-tuning large models (e.g., supervised, RLHF)
Education
- Bachelor's degree in Computer Science, Mathematics, Applied Stats, Machine Learning, or a related field
Benefits
- Healthcare benefits
- 401(k) retirement plan
- Generous PTO
- Equity compensation
- Professional development opportunities