Research Engineer, Pretraining, Deepmind

Google DeepMind

New York, NY, US
Base: $174,000-$252,000; bonus/equity: included; b...
5 years inference-optimized model design experience
Experience working with product teams
Understanding of xla primitives and jax on tpus
This role focuses on inference-optimized model design specifically for Gemini pretraining at Google DeepMind

Job Summary

  • This role focuses on inference-optimized model design specifically for Gemini pretraining at Google DeepMind.
  • Engineers will work across the Large Language Model preparation stack including pretraining, fine-tuning, and serving.
  • The position requires a deep understanding of Accelerated Linear Algebra (XLA) primitives and how JAX code executes on TPUs.

Matching Summary

This role focuses on inference-optimized model design specifically for Gemini pretraining at Google DeepMind.

Salary

Base: $174,000-$252,000; Bonus/Equity: Included; Benefits: Not specified

Skills & Requirements

Must-have

  • 5 years inference-optimized model design experience
  • Experience working with product teams
  • Understanding of XLA primitives and JAX on TPUs

Nice-to-have

  • Active contributor to research community
  • Collaboration with partner universities
  • Publishing papers in AI domains

Key Requirements

  • Bachelor's degree in Computer Science or related field
  • 5 years of experience with inference-optimized model design
  • Equivalent practical experience to formal education

Work Rights

Not specified

Tailored Resume

Cover Letter