Staff Machine Learning Engineer : Platform Intelligence - Apple Maps

Apple
Cupertino, United States of America2026-03-05

About the job

Apple Maps Client is looking for a Staff Machine Learning Engineer to drive the design, development, and deployment of machine learning models optimized for on-device training and inference. Partnering with the Apple Neural Engine team to profile model performance, identify bottlenecks, and push the limits of what's possible on-device. Crafting technical design documents for new ML features is a core part of this role- outlining model architecture choices, performance targets, and deployment strategies. Your work includes building integration code that connects ML models with platform frameworks and APIs. You will lead cross-functional team projects. Beyond individual contributions, you will shape how the team approaches on-device ML. You will establish evaluation frameworks, define quality benchmarks, and write architecture documents that guide the team's direction. You will review code, mentor engineers, and help build a team culture rooted in technical rigor and collaboration.

Responsibilities

Architect and deliver on-device ML solutions that meet strict latency, memory, power, and accuracy requirements across Apple platforms.

Partner with Services teams on model delivery and update mechanisms( OTA model updates, staged rollouts) and define hybrid inference strategies (on-device vs. server-side).

Collaborate cross-functionally with services, platform, and design teams to influence roadmaps, framework capabilities, and user experiences.

Mentor and grow junior and mid-level ML engineers, fostering a culture of technical excellence, curiosity, and inclusive collaboration.

Champion privacy by design — ensuring ML systems uphold Apple's commitment to user privacy through on-device processing, differential privacy, and minimal data collection.

Qualifications

Minimum

Bachelor’s in Computer Science, Machine Learning, Electrical Engineering, or a related field — or equivalent practical experience.

Strong software engineering fundamentals in an object-orient programming language, with emphasis on writing production-grade, testable, and maintainable code.

Experience with Systems Programming (frameworks/libraries/daemons).

7+ years of industry experience in machine learning engineering, with at least 2 years focused on on-device/edge ML deployment.

Strong proficiency in ML frameworks and tool chain such as PyTorch, TensorFlow, Core ML, Foundation Models Framework and MLX.

Proven track record of shipping ML models into production at scale on mobile or embedded platforms.

Preferred

Master’s, or PhD in Computer Science, Machine Learning, Electrical Engineering, or a related field — or equivalent practical experience.

Familiarity with Swift, and Objective-C.

Experience building and operating end-to-end ML pipelines for on-device models — including training, evaluation, conversion, validation, A/B testing, and OTA model delivery.

Familiarity with federated learning, differential privacy, and on-device training/fine-tuning paradigms.