Pre-silicon ML and Compute Framework Engineer, Graphics, Game and ML

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.

Develop and deliver software representing Apple’s roadmap for compute and ML technologies during hardware development.
Design APIs that enable developers to make full use of new hardware features in the simplest way possible.
Partner with cross-functional teams to ensure our software and hardware roadmap includes the state-of-the-art features and performance.
Contribute to the bring-up and verification of new chips and products using our new GPUs.

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

Experience collaborating in large cross functional projects
Familiarity with Apple framework and API design patterns
Hands on experience with ML frameworks (PyTorch, JAX, CoreML)

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.

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $181,100 and $318,400, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant

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

Learn about reasonable accommodations for job applicants

Apple accepts applications to this posting on an ongoing basis.