Staff Machine Learning Engineer – AI/ML Compiler

$174,000 – $261,000 · Posted 8 days ago

San DiegoLast seen 7 days ago

Summary

Design and develop the end-to-end compilation pipeline for Qualcomm AI Hub Workbench, enabling PyTorch and ONNX models to compile into deployable artifacts targeting LiteRT, ONNXRuntime, and QAIRT across Snapdragon SoCs. Own graph optimization, backend dispatch across CPU/GPU/NPU, model catalog validation, and automated CI/CD pipelines for on-device ML deployment. Requires 4+ years of software/hardware/systems engineering experience and proficiency in Python and C++, with preferred expertise in ML compiler infrastructure, MLIR/ONNX/TVM, torch.export, and on-device frameworks.

Data Science AI C++Machine Learning Python Aten Ir Backend Partitioning CI CD Compiler Engineering Git Graph Optimization Litert Lowering Passes Mlir Onnx Onnx Runtime Operator Fusion PyTorch Qairt Qnn Shape Inference Snapdragon Socs Torch Export Tvm