ML Software Engineer

Apple • Seattle, WA

74d

About The Position

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. 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

• B.S. in Computer Science or other numerate subject with 6+ years experience, or M.S. in Computer Science or other numerate subject with 4+ years experience.
• Quality focus - produce reliable, maintainable, deliverable software.
• Comfortable diving deep - working across multiple levels of abstraction.
• Good at handling relationships & communication - collaborate well with colleagues across a wide range of functions.

Nice To Haves

• Familiar with Apple ML stack (ANE, CoreML, MPS/Metal).
• Experience with high-level general distributed ML stack (PyTorch-distributed, NCCL).
• Knowledge of 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 as it emerges from the research community.
• Write performant and scalable frameworks in Swift and C++ for ML inference tasks.
• Integrate inference code into a full service stack.
• Ensure user traffic is served reliably and performantly.
• Develop code that is easy and safe to develop, update, and monitor.

Benefits

• Advancement opportunities to all applicants, including minorities, women, protected veterans, and individuals with disabilities.
• Apple will not discriminate or retaliate against applicants who inquire about, disclose, or discuss their compensation or that of other applicants.
• Apple will consider for employment all qualified applicants with criminal histories in a manner consistent with applicable law.
• Apple is committed to working with and providing reasonable accommodation to applicants with physical and mental disabilities.
• Apple is a drug-free workplace.

What This Job Offers

• Industry: Computer and Electronic Product Manufacturing
• Education Level: Bachelor's degree
• Number of Employees: 5,001-10,000 employees