Finding the best job has never been easier
Apple On-device ML Infrastructure Engineer User Experience APIs United States, California, Cupertino 958970106
31.03.2025
Share
Our group is looking for an ML Infrastructure Engineer, with a focus on ML user experience APIs. The role is responsible for developing new ML model conversion & authoring APIs that will be a part of coremltools (CoreML’s authoring/conversion toolkit). The role also requires interacting with ML authoring frontend (like PyTorch) to capture graphs from the frontend, and efficiently import these graphs into CoreML stack. The conversion and authoring APIs also include feedback APIs, whereby ML engineers can evaluate/visualize the converted models to evaluate functionality/performance.
As an engineer in this role, you will be primarily focused on developing APIs in coremltools to enable ML engineers to efficiently author/convert ML models to CoreML, including any feedback APIs to help them evaluate the CoreML programs. The conversion APIs also include optimizations to enable peak performance on Apple devices, such as quantization, compression, distillation, etc. The role requires a good understanding of ML modeling (architectures, training vs inference trade-offs, etc.), and a good understanding of designing Python APIs and packages. Key responsibilities:Develop APIs in coremltools for ML engineers to efficiently convert models from ML frontends (such as PyTorch, JAX) into CoreML’s model representation.Develop APIs, and tools for ML engineers to evaluate and converted/authored programs.
- Bachelors in Computer Sciences, Engineering, or related discipline.
- Highly proficient in Python programming, some familiarity with C++ is required.
- Proficiency in at least one ML authoring framework, such as PyTorch, TensorFlow, JAX, MLX.
- Strong understanding of ML fundamentals, including common architectures such as Transformers.
- Knowledge of ML optimization techniques (like quantization, distillation, etc.) is a strong plus.
- Experience with any on-device ML stack, such as TFLite, ONNX, etc.
- Experience with designing Python APIs and production deployment of python packages is a strong plus.
- Experience with MLIR/LLVM or any compiler toolchains is a strong plus.
- 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
