Senior Software Engineer, ML Compilers, Edge TPU
Hybrid (Mountain View, CA; Kirkland, WA)$174k - $252kFull-timePosted 3w ago
About the role
Senior Software Engineer on the EdgeTPU Compiler team responsible for building next-generation compiler optimizations to power ML models on EdgeTPU hardware, translating framework-level code into efficient machine instructions, and owning end-to-end feature delivery.
Responsibilities
- Design and build ML compiler optimizations for EdgeTPU hardware, and extend leading authoring frameworks (such as JAX and PyTorch) to enable seamless, high-performance compilation.
- Triage, root-cause, and resolve complex correctness and performance issues encountered when deploying state-of-the-art ML models on EdgeTPU.
- Propose, design, and implement robust compiler features and fixes to address performance bottlenecks and hardware limitations.
- Partner closely with ML model owners to influence model architectures, ensuring they are designed for optimal, efficient execution on EdgeTPU systems.
- Own project execution from end to end, collaborating with cross-functional partners and stakeholders, and managing priorities/deadlines/deliverables for key compiler feature areas.
Qualifications
- Bachelor’s degree or equivalent practical experience.
- 5 years of experience with software development in one or more programming languages (e.g. C++).
- 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.
- 3 years of experience with compilers (optimization, parallelization, etc.).
Skills
Required:C++Software design and architectureCompilers (optimization, parallelization)
Benefits
- Health, dental, vision, life, disability insurance
- Retirement Benefits: 401(k) with company match
- Paid Time Off: 20 days of vacation per year, 6.15 hours per pay period for first five years
- Sick Time: 40 hours/year (69 hours/year in Seattle) including 5 discretionary sick days per instance
- Maternity Leave (Short-Term Disability + Baby Bonding): 28-30 weeks
- Baby Bonding Leave: 18 weeks
- Holidays: 13 paid days per year
Last updated Oct 23, 2026. Interested? Apply on the company site.
More open roles at Google
- Staff Network Security Engineer, Google Enterprise Network3d agoHybrid (Austin, TX; Sunnyvale, CA)
- Group Product Manager, Google Lens AI Shopping3d agoMountain View, CA, USA · Product Management
- UX Program Manager3d agoHybrid (New York, NY; San Bruno, CA) · UX Programs & Operations
- Senior Staff Software Engineer, Android Automotive OS3d agoHybrid (Kraków, Poland; Munich, Germany)
- 3d ago