ML Compiler Engineer - On-Device AI & ML

Nutanix; Santa Clara (CA); On-site; USD 151,000 - 227,000; Full time

Benefits offered by this job

Annual bonus

RSU grants

Benefits package

Job summary

Qualcomm Technologies, Inc. is seeking a Machine Learning Engineer to build and optimize compiler technologies that translate AI models from PyTorch or ONNX into efficient code for CPU, GPU, and NPU devices.

The role collaborates with cross-functional teams to advance ML frameworks, supports on-device deployment, and contributes to the AI hub with graph optimizations and backend compatibility.

Qualifications

• Bachelor's degree or higher in CS/Engineering/IS or related field.
• 2+ years in hardware/software/systems engineering or related work experience.
• Experience with ML infrastructure or AI framework development preferred.

Responsibilities

• Build and maintain ML compiler technologies turning models into efficient device code.
• Contribute to AI hub compiler and ONNX Runtime QNN backends.
• Develop debugging tools with clear diagnostics for teams.

Skills

Python

C++

ML infrastructure

ML compiler concepts

PyTorch export

ONNX/MLIR/TVM

On-device deployment

CI/CD pipelines

Education

Bachelor's degree in CS/Engineering/IS

Master's degree in CS/Engineering/IS

PhD in CS/Engineering/IS

Tools

MLIR

ONNX

TVM

PyTorch

LiteRT

ExecuTorch

ONNXRuntime

Job description

Qualcomm Technologies, Inc. is seeking a Machine Learning Engineer to build and optimize compiler technologies that translate AI models from PyTorch or ONNX into efficient code for CPU, GPU, and NPU devices.

The role collaborates with cross-functional teams to advance ML frameworks, supports on-device deployment, and contributes to the AI hub with graph optimizations and backend compatibility.