Pro Apps ML Ecosystem Engineer, Graphics, Games ML

Summary

Apple’s Metal Pro Apps Ecosystem team in GPU, Graphics and Displays org provides a suite of high-performance data parallel algorithms for developers inside and outside of Apple for iOS, macOS and Apple TV.

Apple’s GPU, Graphics and Displays org provides the graphics software foundation across all of Apple's innovative products, including Mac, iPad, iPhone, and Apple TV. Our responsibilities encompass the frameworks, developer tools, and device drivers for the GPU and Display on Apple’s hardware. MPSGraph provides a suite of high-performance data parallel algorithms for developers inside and outside of Apple to improve the ML inference potential and performance applications on for iOS, macOS and Apple TV.

GGML's Metal Pro Apps and ML Ecosystem team is now seeking a world-class, driven, and dedicated engineer to work alongside developers to integrate and optimize ML workloads for Apple’s ANE and GPU silicon into their macOS and iOS applications. Are you ready to engage with developers to advance their success using Apple’s tools and technologies related to ML?

• Support adoption of GPU-accelerated training by 3rd party Apps and key middleware.
• Performing in-depth analysis of an application’s full stack to drive the best possible performance across hardware families.
• Representing Apple to these 3rd party clients and open source communities, managing the relationships, and driving the engineering work.

Analyze developers products and report on their use of the ML API, recommending optimizations, algorithms, and techniques to improve their applications. Contribute platform enablement and optimisations to key open source projects. Host developers during in-person engineering sessions. You will be a key source for all aspects of ML application development, including engineering, documentation, training, and relationship management.

• GPU compute framework development, maintenance, and optimization.
• Proven programming and problem-solving skills.
• Good understanding of machine learning fundamentals.
• Exemplary written and oral communication skills.

• GPU compute programming models & optimization techniques.
• Machine learning development using one or more ML frameworks (PyTorch, Tensor Flow. or JAX).
• Experience with system level programming and computer architecture.

At Apple, we're not all the same. And that's our greatest strength. We draw on the differences in who we are, what we've experienced and how we think. Because to create products that serve everyone, we believe in including everyone. Therefore, we are committed to treating all applicants fairly and equally. As a registered Disability Confident employer, we will work with applicants to make any reasonable accommodations.

Apple will consider for employment all qualified applicants with criminal backgrounds in a manner consistent with applicable law.

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here - in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.

Learn about accessibility in Apple’s workplace

Role Number: