6 hours ago
Cupertino, CA, USAMid Level / Senior
H1B Sponsor
Responsibilities
- Develop and improve MLIR-based compiler infrastructure for machine learning model compilation.
- Target the Neural Engine, GPU, and CPU to maximize hardware capabilities for ML execution.
- Support model optimization, transformation, execution, debugging, profiling, and analysis across Apple devices.
- Collaborate with model authoring, runtime, and performance teams.
- Write kernels and compiler optimizations for efficient ML model execution.
Requirements
- 3–5 years of experience working on MLIR-based compilers.
- Familiarity with common machine learning model architectures, execution schemes, and operations.
- Familiarity with C++.
- Familiarity with PyTorch or related training frameworks.
- Preferred: familiarity with Swift.
- Preferred: familiarity with programming paradigms for the GPU, CPU, and Neural Engine.
- Preferred: familiarity with writing kernels for ML 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.