Software Engineering Manager, AI/ML
Seattle, WA, USA +2 · Manager · Full-time · Posted Oct 28, 2026
- Location
- Seattle, WA, USA
Kirkland, WA, USA
Sunnyvale, CA, USA - Experience
- 8 years in software development; 5 years in an ML field (including speech/audio, reinforcement learning, or ML infrastructure) and 5 years leading ML design and end-to-end ML optimization; 3 years technical leadership; 2 years people management or team leadership.
- Education
- Bachelor’s degree or equivalent practical experience.
About the job
Leads a team developing and optimizing JAX-based generative AI model frameworks and performance for Google Cloud workloads across TPUs and GPUs.
What you'll do
- Coach and develop engineers working on distributed ML systems, performance optimization, and generative AI models.
- Set and execute the roadmap for JAX-based LLM (MaxText) and Diffusion (MaxDiffusion) frameworks with product and technical partners.
- Guide performance optimization and accelerator support, and collaborate on reusable software and MLPerf submissions.
What they ask for
- Bachelor’s degree or equivalent practical experience.
- 8 years of experience in software development.
- 5 years of experience in an ML field and 5 years leading ML design and optimizing end-to-end Machine Learning.
- 3 years in a technical leadership role and 2 years in people management or team leadership.
Nice to have
- Master’s degree or PhD in Engineering, Computer Science, or a related technical field.
- 3 years working in a complex, matrixed organization involving cross-functional or cross-business projects.
Pay
USD $207000 - $300000 (base salary)
Tools
JAX, MaxText, MaxDiffusion, TPUs, GPUs, GCP AI Hypercomputer, MLPerf
AI focus
Leads development and performance optimization of LLM and diffusion frameworks for generative AI workloads, including quantization, sparsity, custom kernels, and support for new accelerator architectures.
Model-written summary · the original posting is authoritative