Skip to content
On-device ML Infrastructure Engineer – ML User Experience Apis
| Company | Apple |
|---|
| Location | Cupertino, CA, USA |
|---|
| Salary | $143100 – $264200 |
|---|
| Type | Full-Time |
|---|
| Degrees | Bachelor’s |
|---|
| Experience Level | Junior, Mid Level |
|---|
Requirements
- 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.
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 in coremltools for ML engineers to author/tailor programs to achieve peak performance on Apple devices (e.g., quantization, distillation, custom operations, etc.)
- Develop APIs, and tools for ML engineers to evaluate and converted/authored programs.
Preferred Qualifications
- 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.
