On-Device ML Systems & Performance Engineer

Socket.dev; Cupertino (CA); On-site; USD 140,000 - 200,000; Full time

Job summary

Apple is hiring an ML Infrastructure Engineer to advance on-device machine learning workloads. You will help build the core runtime and orchestration tools that enable efficient model execution across Apple devices, collaborating with compiler and runtime teams to maximize stability and performance.

The role emphasizes transforming research into practical, scalable infrastructure, with a focus on embedding ML capabilities into the device ecosystem and optimizing for diverse hardware constraints.

Qualifications

• 3-5 years of tooling experience in Python 3 and C++/Swift.
• Familiarity with common ML model architectures and execution schemes.
• Experience with PyTorch or related frameworks.

Responsibilities

• Develop the end-to-end on-device ML runtime and tooling.
• Collaborate with model authoring, compiler, and runtime teams to optimize performance.
• Contribute to model orchestration and efficient execution across devices.

Skills

Python 3

C++/Swift

ML architectures

PyTorch

Tools

MLIR

Job description

Apple is hiring an ML Infrastructure Engineer to advance on-device machine learning workloads. You will help build the core runtime and orchestration tools that enable efficient model execution across Apple devices, collaborating with compiler and runtime teams to maximize stability and performance.

The role emphasizes transforming research into practical, scalable infrastructure, with a focus on embedding ML capabilities into the device ecosystem and optimizing for diverse hardware constraints.