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.

Staff Software Engineer, Machine Learning Compilers, Edge TPU

Job Description