2 months ago
Cupertino, CA, USAMid Level / Senior
H1B sponsor
Responsibilities
- Develop and improve core runtime infrastructure for machine learning model execution across Apple devices.
- Build model orchestration capabilities focused on execution stability and performance.
- Collaborate with model authoring, compiler, and runtime teams.
- Support model optimization, transformation, debugging, profiling, benchmarking, and analysis workflows.
- Work on ML execution across CPU, GPU, Neural Engine, embedded systems, and larger compute environments.
- Write kernels and contribute to tooling for efficient ML model execution.
Requirements
- 3–5 years of experience working on tooling built in Python 3 and C++ or Swift.
- Familiarity with common machine learning model architectures, execution schemes, and operations.
- Familiarity with PyTorch or related training frameworks.
- Preferred: experience working on or adjacent to MLIR-based compilers.
- Preferred: familiarity with deploying applications or tooling on Apple platforms.
- Preferred: familiarity with programming paradigms for GPUs, CPUs, and Neural Engine hardware.
- Preferred: familiarity with writing kernels for machine learning model execution.
Categories
About Apple
Apple designs and sells consumer electronics, software, and services for consumers and professionals worldwide, including iPhone, Mac, iPad, Apple Watch, and AirPods, plus platforms like iOS/macOS and services such as the App Store, iCloud, Music, and TV+. Its business combines device sales with services and subscriptions and in-house silicon design. Founded in 1976, Apple is headquartered in Cupertino, California, and trades on NASDAQ as AAPL.