Apple’s Graphics, Games, and Machine Learning team provides the compute and graphics software foundation across all of our innovative products including iPhone, iPad, Apple TV, Mac, Vision Pro, and Apple Watch.
The Pre-silicon Compute Frameworks team is seeking a senior engineer to influence and lead new features in Metal and Metal compute frameworks through our next-generation GPU hardware roadmap.
The Pre-silicon Compute Frameworks team is responsible for API design and development across Metal and Metal compute frameworks for upcoming hardware features in future Apple Silicon GPUs.
As a member of this team, you will contribute to API design, development, and performance analysis across Metal and Metal compute frameworks for future Apple Silicon GPUs. Along with API development, you will develop critical compute and ML use cases to showcase new features.
You will partner with other leaders across Apple software and hardware organizations to propose, develop, and evaluate hardware features for new silicon generations from emulation through silicon.
Experience collaborating in large cross functional projects
Familiarity with Apple framework and API design patterns
Hands on experience with ML frameworks (PyTorch, JAX, CoreML)
BS/MS/PhD or equivalent in Computer Science or related field with 5 years experience.
Strong grasp and practical experience with at least one GPU compute APIs (Metal, CUDA, OpenCL, etc.)
Good understanding of machine learning fundamentals and familiarity with ML frameworks (PyTorch, JAX, CoreML)
Knowledge of Operating Systems, Computer Architecture, and CPU/GPU programming
Strong background in C/C++ programming
We’re a diverse collective of thinkers and doers, continually reimagining what’s possible to help us all do what we love in new ways. And the same innovation that goes into our products also applies to our practices — strengthening our commitment to leave the world better than we found it. This is where your work can make a difference in people’s lives. Including your own.
Apple is an equal opportunity employer that is committed to inclusion and diversity. Visit apple.com/careers to learn more.
