Learn more about Google. Explore our innovative AI products and services, and how we. Backed by Kleiner Perkins.
About the role
Our work powers Gemini and Gemma by developing core reasoning capabilities and RL scaling for Gemini 3, and leading Gemma 270M, including multiagent Gemini capabilities. We deliver critical contributions to AI Grand Challenges such as our gold medal winning IMO 2025 effort, drive product innovations like deep think mode and agentic inference scaling in antigravity, and lead Alphabet wide initiatives including AI for Science and Project Big Sleep.
What they're looking for
- Bachelor's or Master's degree in Computer Science, Mathematics, Physics, a related quantitative field, or equivalent practical experience
- 4 years of experience building, scaling, and debugging machine learning models using deep learning frameworks (e.g., JAX, PyTorch, or TensorFlow)
More about this role
- Operate across the full research-and-engineering lifecycle of frontier reasoning and agentic systems.
- Address unsolved problems in agentic reasoning, turning early exploratory prototypes into hardened production features for Gemini releases.
- Architect and optimize distributed post-training pipelines and agent-environment simulation loops across thousands of accelerators.
- Design rigorous experiments and failure analyses to isolate performance bottlenecks and communicate findings through clear write-ups.
- Drive technical excellence by maintaining high code quality and architectural health across shared reinforcement learning and modeling codebases.
- Bachelor's or Master's degree in Computer Science, Mathematics, Physics, a related quantitative field, or equivalent practical experience.
- 4 years of experience building, scaling, and debugging machine learning models using deep learning frameworks (e.g., JAX, PyTorch, or TensorFlow).
Experience in at least one core area: Reinforcement Learning (RL), Post-Training (SFT/RLHF/RLAIF), Agentic Tool-Use, or Inference-Time Search.
- PhD in Computer Science, Machine Learning, Physics, or a related quantitative field.
- Experience...
Browse similar: AI jobs · AI startup jobs · Startup jobs