On-Device ML Infrastructure Engineer (CoreML Runtime), Graphics, Games and Machine Learning
Apple5 days ago
Cupertino, CA, USAMid Level
H1B Sponsor
Responsibilities
- Build and optimize Core ML graph compilation and runtime systems for efficient on-device model execution.
- Develop compiler optimizations, runtime components, operator primitives, and system software for Apple products.
- Support model authoring, transformation, execution, debugging, profiling, benchmarking, and analysis across the ML development workflow.
- Enable machine learning models to run efficiently on resource-constrained Apple devices and accelerators.
Requirements
- Master’s degree or equivalent experience in Computer Science, Engineering, or a related field.
- High proficiency in C++ or Swift and familiarity with Python.
- Experience with a compiler stack such as MLIR, LLVM, or TVM.
- Familiarity with operating systems, embedded programming, and parallel programming.
- Sound understanding of machine learning fundamentals and architectures such as Transformers.
- Strong communication skills for working with multifunctional audiences.
- Preferred experience with on-device ML stacks such as TFLite, ONNX, or ExecuTorch.
- Experience with ML authoring frameworks such as PyTorch, TensorFlow, or JAX is preferred.
- Experience with accelerators or GPU programming is preferred.
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.