Staff Software Engineer, Machine Learning Compilers, Edge Tpu

Google

Mountain View, CA, USA
Base: $207,000-$300,000; bonus/equity: included; b...
Not specified; likely hybrid or onsite.
Machine learning compiler development
Edgetpu hardware optimization
Jax and pytorch frameworks
Google is seeking a Staff Software Engineer for its EdgeTPU compiler team, focusing on machine learning compilers and optimization for EdgeTPU hardware. The ideal candidate should possess expertise in ML frameworks and a passion for pushing technological boundaries

Matching Summary

Match Score: 85

Google is seeking a Staff Software Engineer for its EdgeTPU compiler team, focusing on machine learning compilers and optimization for EdgeTPU hardware. The ideal candidate should possess expertise in ML frameworks and a passion for pushing technological boundaries.

Salary

Base: $207,000-$300,000; Bonus/Equity: Included; Benefits: Included

Skills & Requirements

Must-have

  • Machine Learning Compiler Development
  • EdgeTPU Hardware Optimization
  • JAX and PyTorch Frameworks
  • ML Runtime Systems Deployment
  • Hardware/Software Interface Design

Nice-to-have

  • Leadership qualities in fast-paced environment
  • Full-stack versatility
  • Experience with distributed computing
  • Background in large-scale system design

Key Requirements

  • Staff Software Engineer level experience
  • Expertise in ML compiler analysis and optimization
  • Proficiency with JAX and PyTorch frameworks

Work Rights

Not specified

Tailored Resume

Cover Letter