Senior Software Engineer, ML Compilers, Edge TPU

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 systematically 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.
• - 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 systematically 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.

Minimum qualifications:

Preferred qualifications:

Qualifications

• Minimum 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.). Preferred qualifications: - Master's degree or PhD in Computer Science or a related technical field. - Compiler development experience in the context of accelerator-based architectures. - Experience working with low-level software or strong interest in developing low-level software. - Experience working with hardware such as CPU, GPU, or TPU. - Experience in optimizing ML model inference on device.