Senior Software Engineer, Gemini Audio, DeepMind
Mountain View, CA, USA · Senior
Sponsorship not specified$174k-$253kDetected 13 hours ago
AlgorithmsMachine LearningPyTorchResearch
About the role
- We use our technologies for widespread public benefit and scientific discovery, ensuring safety and ethics are always our highest priority.
- Our global teams offer diverse learning opportunities and varied career pathways for those driven to achieve exceptional results through collective effort.
Responsibilities
- Design, build, and maintain training infrastructure to support Gemini Audio encoder and pretraining models, focusing on Accelerated Linear Algebra (XLA) optimization for training on Tensor Processing Units (TPUs).
- Contribute and work closely with Gemini to build and maintain audio related components.
- Collaborate with research teams across Gemini Audio to improve the model training and inference efficiency.
Requirements
- Bachelor's degree or equivalent practical experience.
- 5 years of experience with software development in Python.
- 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.
- Experience with compiler optimization, code generation, and runtime systems for popular accelerators, including GPU or TPU.
Nice to have
- Master's degree or PhD in Computer Science or related technical field.
- Experience with TPU architectures, memory hierarchies, performance bottlenecks, and Accelerated Linear Algebra.
- Experience tailoring algorithms and ML models to exploit TPU architecture strengths and minimize weaknesses.
- Experience with various stack layers: compilers (OpenXLA, MLIR), serving libraries/frameworks (vLLM, sglang) and ML frameworks (JAX, PyTorch).
- Debugging experience to improve performance of single-mode or multi-mode (distributed) systems.
- Artificial intelligence will be one of humanity's most transformative inventions.
Compensation
- US: $174000 - $253000 (USD) + 15% bonus target + equity + benefits
Benefits
- Implement tools to analyze and track audio model training efficiency and health.
This listing is sourced directly from Google's careers page and normalized into a canonical job model.