Pro Apps ML Ecosystem Engineer

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. 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 to improve ML inference potential and performance applications on iOS, macOS, and Apple TV.

Description

GGD SW's Metal Pro Apps 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?

Responsibilities:

1. Analyze developers' products and report on their use of the ML API, recommending optimizations, algorithms, and techniques to improve their applications.
2. Host developers during in-person engineering sessions.
3. Support adoption of GPU-accelerated training by 3rd party Apps.
4. Perform in-depth analysis of an application’s full stack to drive the best possible performance across hardware families.
5. Represent Apple to these 3rd party clients, managing the relationships, and driving the engineering work.

Minimum Qualifications

• Proven programming and problem-solving skills.
• Good understanding of machine learning fundamentals.
• GPU compute programming models & optimization techniques.
• GPU compute framework development, maintenance, and optimization.
• Machine learning development using one or more ML frameworks (TensorFlow, PyTorch, or JAX).
• Experience with system level programming and computer architecture.

Preferred Qualifications

• You must be enrolled as a student on a Masters, PhD, or equivalent course.