Staff Software Engineer, Inference Performance Optimization, GenAI
Google DeepMind · Mountain View, CA, USA
CHECKED 4M AGO
Frontier labonsiteStaff$207k–$300k
At a glance
Highlights
- Push boundaries of large-scale AI inference to make it faster and cheaper.
- Analyze full inference stack to find bottlenecks and drive system improvements.
- Build tools tracking compute spend to guide latency and cost tradeoffs.
What they offer
- 20% bonus target
- equity
- benefits
Ideal candidate
Systems-driven engineer with deep performance intuition who thrives profiling complex distributed AI workloads and turning first-principles insights into faster, cheaper inference at scale.
Requirements
- Bachelor's degree in Computer Science, Computer Engineering, Electrical Engineering, Applied Mathematics, or related technical field, or equivalent practical experience
- 8 years of experience in software development
- Experience in Python and C++, including navigating, debugging, and modifying serving codebases
- Experience with AI model execution constraints, throughput-latency tradeoffs, memory bandwidth limitations, and modern serving architectures
- Experience with real-world LLM inference serving environments or contributions to open-source inference frameworks (e.g., vLLM, TensorRT-LLM, SGLang, Dynamo)
- Experience profiling workloads using standard ML profilers (e.g., PyTorch profiler) and internal trace analysis tools
- Experience with observability and reliability for large distributed systems
- Familiarity with GPU/TPU/accelerator performance concepts and implications for inference serving performance
Lab profile
About Google DeepMind
Google's AI lab building frontier models like Gemini to advance science and general intelligence.
What they build
- Gemini
- Gemma
- AlphaFold
Known for
- Frontier foundation models and GenAI
- Large-scale training with JAX, PyTorch, TensorFlow
- Model evaluation and deployment expertise
- Responsible frontier AI research
Key facts
- Founded in 2010
- Headquartered in London, UK
- Part of Google (Alphabet)