Job Overview
Work side-by-side with frontier model researchers and flagship enterprise partners to deploy multimodal Gemini systems on Cloud TPU clusters.
Key Responsibilities
- Port research breakthrough model checkpoints into scalable production inference pipelines.
- Optimize million-token context window retrieval and video-audio multimodal understanding.
- Benchmark and profile distributed JAX workloads across massive TPU v5p pods.
Requirements
- Expertise in Python, JAX/PyTorch, and distributed machine learning.
- Track record building high-reliability production systems or publishing at top AI venues (NeurIPS, ICML).