Job description

We’re building an end-to-end developer experience for machine learning development that leverages Apple’s vertical integration. This allows developers to iterate on model authoring, optimization, transformation, execution, debugging, profiling, and analysis. This role focuses on the core runtime for execution across a wide variety of devices and use cases. We’re seeking a highly motivated software engineer who is creative, talented, and passionate about machine learning, common compiler optimizations, and system software engineering in the fast-paced and dynamic field of machine learning. 3-5 years working on tooling built in Python 3 and C++/Swift Familiarity with common ML model architectures, execution schemes, and operations. Familiarity with PyTorch or related training frameworks. Experience working on or adjacent to MLIR-based compilers. Familiarity with deploying applications or tooling on Apple platforms. Familiarity with programming paradigms for the GPU, CPU, and Neural Engine. Familiarity with writing kernels for ML model execution.

Requirements

Must-haves

3-5 years working on tooling built in Python 3 and C++/Swift
Familiarity with common ML model architectures, execution schemes, and operations.
Familiarity with PyTorch or related training frameworks.
Experience working on or adjacent to MLIR-based compilers.
Familiarity with deploying applications or tooling on Apple platforms.
Familiarity with programming paradigms for the GPU, CPU, and Neural Engine.
Familiarity with writing kernels for ML model execution.

Nice-to-haves

None flagged.

Application prep

Use the role’s language and application details while tailoring your materials.

Keywords to mirror

machine learningpythonswiftml

Platform: apple
Effort: medium