6 hours ago
Cupertino, CA, USAMid Level / Senior
H1B Sponsor
Responsibilities
- Develop and improve ML compiler infrastructure focused on model compilation and runtime execution.
- Use the MLIR-based compiler stack to target Apple’s Neural Engine, GPU, and CPU.
- Collaborate with model authoring, runtime, and performance teams to maximize hardware utilization.
- Support model optimization, transformation, execution, debugging, profiling, and analysis across Apple devices.
- Write or improve kernels for machine learning 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 GPU, CPU, and Neural Engine environments.
- 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.