Machine Learning Engineer – Edge AI & On-Device Optimization
Referral available at Apple
An employee here can refer you. The referrer stays anonymous and reaches out to you directly if interested.
Join our team as a Machine Learning Engineer and help shape the future of on-device AI. You'll research, design, and deploy cutting-edge deep learning models optimized for Apple silicon edge devices, working across the full ML lifecycle alongside hardware, software, and product teams.
We are looking for a talented and motivated Machine Learning Engineer to join our team. You will work within a collaborative, research-driven engineering culture that values innovation and rigor, with the opportunity to build impactful AI products deployed at scale on real devices. We offer competitive compensation, benefits, and opportunities for professional growth.
Minimum Qualifications
- M.Sc. or Ph.D. in Computer Science, Electrical Engineering, or a related field - or equivalent practical experience
- Strong foundation in deep learning theory and hands-on experience training large-scale models
- Proficiency in Python and deep learning frameworks such as PyTorch or TensorFlow/JAX
- Hands-on experience with model compression and optimization techniques (quantization, pruning, distillation, etc.)
- Familiarity with on-device inference frameworks such as Core ML, TensorFlow Lite, ONNX Runtime, or TensorRT
- Experience working with multimodal data (e.g., images, audio, time-series, or sensor fusion)
- Strong analytical and problem-solving skills; ability to translate research ideas into production-quality code
Preferred Qualifications
- Experience deploying models to embedded systems, mobile devices, or custom silicon (NPU/DSP)
- Familiarity with hardware-aware neural architecture search (NAS) or AutoML techniques
- Exposure to low-level optimization techniques such as mixed-precision training or operator fusion
- Hands-on experience with Apple Neural Engine and Core ML for on-device inference
- Publications or open-source contributions in efficient deep learning or edge AI
- Experience with MLOps workflows and CI/CD pipelines for model development