Staff Machine Learning Engineer – AI/ML Compiler

Summary

Staff Machine Learning Engineer role focused on designing and maintaining Qualcomm AI Hub's end-to-end ML compilation pipeline, from PyTorch and ONNX model ingestion through graph optimization to deployable artifacts on Snapdragon SoCs. Responsibilities span compiler infrastructure (graph transformation, op validation, backend dispatch across CPU/GPU/NPU), model catalog automation and CI/CD, and developer tooling for profiling and debugging. Requires 4+ years of software/systems engineering experience (or 3+ with a master's, or 2+ with a PhD), with preferred expertise in ML compiler concepts, ONNX/PyTorch export workflows, and on-device deployment frameworks.