Back to jobs
Job Description
- 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.