6 hours 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
We’re a diverse collective of thinkers and doers, continually reimagining what’s possible to help us all do what we love in new ways. And the same innovation that goes into our products also applies to our practices — strengthening our commitment to leave the world better than we found it. This is where your work can make a difference in people’s lives. Including your own. Apple is an equal opportunity employer that is committed to inclusion and diversity. Visit apple.com/careers to learn more.