ML Software Engineer

Apple • Seattle, WA

12h • Remote

About The Position

People at Apple don't just build products — they craft the kind of experience that has revolutionized entire industries. The diverse collection of our people and their ideas inspire innovation in everything we do. Our team builds ML-inference applications and services on Apple Silicon in the datacenter, specifically focusing in recent years on generative AI as part of the Private Cloud Compute component of Apple Intelligence. Join us. Be you. As part of the team you will help engineer continuous improvements in stability and performance for private cloud compute, as well as help implement entirely new functionality as it emerges from the research community, in collaboration with product teams throughout Apple. We write performant and scalable frameworks (in Swift and C++) to distribute and coordinate ML inference tasks to different hardware acceleration IP blocks on different SoCs. We’re a collection of highly skilled and friendly engineers who value each other’s opinions and experience. We strive for excellence and believe strongly in the quality of our output. We have formed a team of domain experts who specializes in specific core subject areas, and also have broad experience of cloud software services and platforms. You will integrate inference code into a full service stack to ensure that user traffic is served reliably and performantly, and will have a strong focus on developing code that is easy and safe to develop, update and monitor.

Requirements

• Experience working as a software engineer on large production systems.
• Experience programming in Swift, C, C++.
• Experience with iOS/macOS or Xcode.
• Practical experience running machine learning models.
• Experience evaluating machine learning models for quality and performance metrics.

Nice To Haves

• Familiar with Apple ML stack (ANE, CoreML, MPS/Metal).
• Familiar with high-level general distributed ML stack (PyTorch-distributed, NCCL).
• Familiar with high throughput inter-chip communication systems.
• On-device iOS development experience.

Responsibilities

• Engineer continuous improvements in stability and performance for private cloud compute.
• Implement new functionality emerging from the research community, in collaboration with product teams.
• Write performant and scalable frameworks (in Swift and C++) to distribute and coordinate ML inference tasks.
• Integrate inference code into a full service stack to ensure reliable and performant user traffic serving.
• Develop code that is easy and safe to develop, update, and monitor.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed