Staff ML Engineer, On-Device Platform Intelligence

Apple Inc.; Cupertino, Northern (CA, KY); Hybrid; USD 185,000 - 325,000; Full time

Benefits offered by this job

RSU/Bonuses

Employee stock plan

Medical & dental coverage

Tuition reimbursement

Relocation support

Job summary

Apple Inc. in Cupertino, CA is seeking a Staff Machine Learning Engineer to lead the design, development, and deployment of machine learning models optimized for on-device training and inference.

You will partner with the Apple Neural Engine team to profile performance, draft architecture documents, and drive deployment strategies across platforms. You will mentor engineers, shape team practices for on-device ML, review code, and help build a culture grounded in rigor and collaboration.

Qualifications

• Bachelor’s in Computer Science, Machine Learning, Electrical Engineering, or related field — or equivalent practical experience.
• Strong software engineering fundamentals with 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 on-device/edge ML deployment.
• Strong proficiency in ML frameworks and tool chain (PyTorch, TensorFlow, Core ML, Foundation Models Framework, MLX).
• Proven track record of shipping ML models into production at scale on mobile/embedded platforms.

Responsibilities

• Architect and deliver on-device ML solutions meeting latency, memory, power, and accuracy constraints.
• Collaborate on model delivery mechanisms and define hybrid inference strategies.
• Cross-functional collaboration to influence roadmaps, frameworks, and user experiences.
• Mentor and grow junior/mid-level ML engineers and foster technical excellence.
• Champion privacy by design with on-device processing and differential privacy.

Skills

Machine learning engineering

On-device ML

PyTorch

TensorFlow

Core ML

Foundation Models Framework

MLX

Production ML on mobile

Software engineering fundamentals

Systems programming

Education

Bachelors in CS/ML/EE

Masters/PhD preferred

Tools

Swift

Objective-C

Job description

Apple Inc. in Cupertino, CA is seeking a Staff Machine Learning Engineer to lead the design, development, and deployment of machine learning models optimized for on-device training and inference.

You will partner with the Apple Neural Engine team to profile performance, draft architecture documents, and drive deployment strategies across platforms. You will mentor engineers, shape team practices for on-device ML, review code, and help build a culture grounded in rigor and collaboration.