Work as part of the EdgeTPU compiler team, building ML compilers for EdgeTPU hardware and analyzing and improving the compiler quality and performance on optimization decisions, correctness and compilation time.

Work with and extend ML authoring frameworks, including JAX, Pytorch to compile ML models for the EdgeTPU.

Work with ML runtime systems to deploy optimized ML models on the EdgeTPU.

Work with EdgeTPU architects to design the Hardware/Software (HW/SW) interface, and co-optimizations between CPU, GPU, and TPU.

Collaborate with ML model developers, researchers, and EdgeTPU hardware/software teams to accelerate the transition from research ideas to user experiences running on the EdgeTPU.

Bachelor’s degree or equivalent practical experience.

8 years of experience in software development.

5 years of experience testing, and launching software products, and 3 years of experience with software design and architecture.

5 years of experience with Machine Learning compilers (optimization, parallelization, etc.).

5 years of experience with relevant ML design and ML infrastructure (e.g., model deployment, model evaluation, etc.).