Internship - ML Pro Apps

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 inside and outside of Apple to improve the 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

• Good written and oral communication skills.
• Machine learning development using one or more ML frameworks (TensorFlow, PyTorch, or JAX).
• A computer science or related degree is preferred. Alternative degrees that would also be acceptable would be ML programming or games programming.

Preferred Qualifications

• Good understanding of machine learning fundamentals.