Software Engineer III, Performance
Sunnyvale, CA, USA · Full-time · Posted Oct 7, 2026
- Experience
- 2 years in each of the three areas stated in the minimum qualifications.
- Education
- Bachelor’s degree or equivalent practical experience.
About the job
Develop and productionize algorithmic and kernel optimizations for AI model performance, while collaborating on hardware-software co-design for future TPU capabilities.
What you'll do
- Track AI models, frameworks, and research to identify optimization opportunities, then validate optimizations with internal customers.
- Transition validated algorithms and recipes to performance and automation teams for broader scaling.
- Develop and productionize custom Pallas kernels, and collaborate on future TPU architecture and product capabilities.
What they ask for
- Bachelor’s degree or equivalent practical experience.
- 2 years of experience with software development in one or more programming languages.
- 2 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging.
- 2 years of experience with computer architecture, performance analysis, and performance modeling.
Nice to have
- Master's degree or PhD in Computer Science or related technical fields; experience with data structures and algorithms.
- Experience with hardware-software algorithmic co-design and computer architecture analysis; experience developing accessible technologies; strong background in ML models and systems.
Pay
$147000 - $210000 (USD) base salary; 15% bonus target + equity + benefits
Tools
Pallas, TPU, mxfp4, nvfp4
AI focus
Evaluate AI models, frameworks, and research; identify and validate algorithmic optimizations such as quantization; develop custom kernels for model efficiency.
Model-written summary · the original posting is authoritative