On-Device ML Graph Compiler Engineer

Apple; Cupertino (CA); On-site; USD 120,000 - 160,000; Full time

Job summary

Apple is seeking an ML Infrastructure Engineer to join the On-Device Machine Learning team. You will work on building innovative ML graph compilation and runtime systems that optimize machine learning models efficiently on Apple products. This role is crucial for enhancing the user experience across various applications.

The ideal candidate should possess strong expertise in C++ or Swift, with additional knowledge of Python, and have a passion for machine learning fundamentals. Join us to shape the future of intelligent experiences on Apple platforms.

Qualifications

• Masters or equivalent experience in Computer Sciences, Engineering, or related subject area.
• Highly proficient in C++ or Swift. Familiarity with Python.
• Experience with any compiler stack (MLIR/LLVM/TVM/...).

Responsibilities

• Architect and maintain the on-device graph compiler and runtime.
• Develop production-critical system software for implementing ML models.
• Optimize model execution for performance, energy efficiency, and thermal management.

Skills

C++

Swift

Python

Compiler stacks

Communication skills

Education

Masters or equivalent experience in Computer Sciences, Engineering

Tools

MLIR

LLVM

TVM

PyTorch

TensorFlow

Job description

Apple is seeking an ML Infrastructure Engineer to join the On-Device Machine Learning team. You will work on building innovative ML graph compilation and runtime systems that optimize machine learning models efficiently on Apple products. This role is crucial for enhancing the user experience across various applications.

The ideal candidate should possess strong expertise in C++ or Swift, with additional knowledge of Python, and have a passion for machine learning fundamentals. Join us to shape the future of intelligent experiences on Apple platforms.