Staff Machine Learning Software Engineer

Summary

Staff-level machine learning software engineer responsible for designing, implementing, and optimizing ML software for embedded systems including mobile, edge, auto, and IoT products running on ARM, DSP, and GPU processors. Primary responsibilities include developing and optimizing ML kernels and frameworks (TensorFlow, PyTorch) for efficient hardware execution, implementing performance-critical software in C++/Python, and collaborating with hardware teams on joint design and optimization. Will lead complex projects involving integration, testing, and tool development for embedded ML subsystems, coordinate cross-team dependencies, and contribute to design reviews and technical planning.