356 jobs for "Lead Machine Learning Engineer"

Every iPhone, iPad, Mac, Apple Watch, Vision Pro, and Apple TV runs on the compute and graphics software foundation built by Apple's Graphics, Games, and Machine Learning team. The Pre-silicon Compute Frameworks team sits at the leading edge of that work designing the Metal and Metal compute frameworks and validating performance for features that will define future Apple Silicon GPUs.

We're looking for an engineering manager with a strong technical background who's excited to build and lead a team of engineers, and to partner with software and hardware leaders to shape Apple's GPU compute roadmap!

Description

As manager of the Pre-silicon Compute Frameworks team, you own the new APIs for Apple's next GPU generations. You will lead, support, and grow a team of engineers building Metal and Metal compute framework features and conducting performance analysis on next generation GPU hardware, carrying features from emulation all the way through silicon and into the bring-up of new chips and products.

You will partner directly with leaders across Apple's software and hardware organizations to propose, develop, and evaluate features for new silicon generations, representing Apple's compute and ML roadmap while the hardware is still taking shape.","responsibilities":"Lead, support, and grow the Pre-silicon Compute Frameworks team to deliver software that represents Apple's roadmap for compute and ML technologies during hardware development

Guide the team's API designs so developers can take full advantage of new hardware features in the simplest way possible

Partner with cross-functional teams to ensure the software and hardware roadmap aligns with state-of-the-art features and performance

Contribute to the bring-up of new chips and products built on our new GPUs

Preferred Qualifications

Experience defining, planning, and executing complex projects involving partnership across multiple teams

Track record in strategic leadership and decision-making

Familiarity with Apple framework and API design patterns

Hands-on experience with ML frameworks such as PyTorch, JAX, or CoreML

Minimum Qualifications

BS/MS/PhD in Computer Science, or a related field; or equivalent experience

2-3 years of experience as a manager or lead

Experience with at least one GPU compute API (Metal, CUDA, OpenCL, or similar)

Understanding of machine learning fundamentals, and familiarity with ML frameworks such as PyTorch, JAX, or CoreML

Knowledge of operating systems, computer architecture, and CPU/GPU programming, with experience in C/C++

Pay & Benefits

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 $206,200 and $356,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.

About Apple

Working at Apple means doing more than you ever thought possible and having more impact than you ever imagined.

Size: 10,000+ employees
Industry: Technology

View company profile