Staff Research Scientist: ML Architect & Evaluator

Google LLC; Mountain View (CA); Presencial; USD 207 000 - 300 000; Tempo integral

Vantagens oferecidas por esta oferta de emprego

Equity

Annual bonus

Resumo da oferta

Google DeepMind in Mountain View, CA, seeks a Staff Research Scientist for Gemini Research. You will invent ML architectures, optimizers, and training methods, and build systems to test ideas at scale across distributed GPUs/TPUs.

This is a deep individual-contributor role requiring a PhD or equivalent, strong publication record, and leadership across teams. You will work with JAX/Flax and PyTorch, collaborating with pre-training, post-training, and infrastructure teams to land improvements into

Qualificações

• PhD in Computer Science, a related field, or equivalent practical experience.
• 4 years of experience with research agendas across multiple teams or projects.
• One or more scientific publication submission(s) for conferences, journals, or public repositories.
• Experience working with deep learning frameworks (e.g., JAX/Flax, PyTorch) for distributed systems and accelerator clusters.

Responsabilidades

• Invent, implement, and scale new Machine Learning (ML) algorithms, such as architectures, optimizers, training objectives, and recipes.
• Design and build automated research systems in which Gemini proposes, codes, and evaluates changes to its own training stack, moving toward a model that designs its successor.
• Serve as the final verifier for AI-proposed algorithm changes: reproduce results, design decisive ablations across model scales, rule out confounds such as evaluation leakage or unequal tuning budgets, and decide what moves forward.
• Build and own lean, high-performance training and evaluation code (JAX on Tensor Processing Units, with custom kernels where needed) allowing researchers and AI agents can test ideas at meaningful scale.
• Partner with Gemini pre-training, post-training, and infrastructure teams to land validated improvements in production models, and share results through internal talks.

Conhecimentos

Research leadership

Deep learning frameworks

Publications

Distributed systems

Formação académica

PhD in Computer Science or related field

Ferramentas

JAX/Flax

PyTorch

Descrição da oferta de emprego

Google DeepMind in Mountain View, CA, seeks a Staff Research Scientist for Gemini Research. You will invent ML architectures, optimizers, and training methods, and build systems to test ideas at scale across distributed GPUs/TPUs.

This is a deep individual-contributor role requiring a PhD or equivalent, strong publication record, and leadership across teams. You will work with JAX/Flax and PyTorch, collaborating with pre-training, post-training, and infrastructure teams to land improvements into