Finding the best job has never been easier
Apple On-device ML Infrastructure Engineer Compiler & Runtime United States, California, Cupertino 733579438
03.04.2025
Share
Key responsibilities:Ensure our infrastructure can run optimally for a wide range of first and third party machine learning models.
- Bachelors in Computer Science, Engineering, or related discipline.
- Highly proficient in C++. Familiarity with Python and/or Swift
- Familiarity with Operating Systems and Embedded Programming.
- Sound understanding of ML fundamentals, including common architectures such as Transformers.
- Experience with any on-device ML stack, such as TFLite, ONNX, ExecuTorch, etc.
- Experience with open source machine learning models (Mistral, Phi, Gemma, Huggingface, etc)
- Experience with any compiler stack (MLIR/LLVM/TVM/...).
- Experience with any ML authoring framework (PyTorch, TensorFlow, JAX, etc.).
- Experience with machine learning accelerators and GPU programming.
- Good communication skills, including ability to communicate with cross-functional audiences.
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.
These jobs might be a good fit
