Job Description
Summary
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.
Description
Responsibilities:
* Support adoption of GPU-accelerated training by 3rd party Apps.
* 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, 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