Server ML Frameworks Engineer

Apple, Inc.; Cupertino (CA); On-site; USD 185,000 - 325,000; Full time

Benefits offered by this job

Employee stock purchase plan

Medical and dental coverage

Retirement benefits

Job summary

Apple is seeking a senior technical leader for the Server ML Frameworks team in Cupertino to define architectures and drive high-throughput GPU work for GenAI models in data-center scale. You will mentor engineers, collaborate with hardware teams, and ship production-grade frameworks and APIs (Metal, CUDA) while shaping the Apple Intelligence compute roadmap across platforms.

You will mentor engineers, collaborate with hardware teams, and ship production-grade frameworks and APIs (Metal, CUDA)

Qualifications

• BS/MS/PhD in Computer Science or a related field, or equivalent practical experience.
• 3–5 years of experience as a manager or technical leader.
• Experience with GPU kernel development and optimization using compute programming models such as Metal, CUDA, or similar.
• Experience with distributed training or inference techniques.
• Strong system-level programming skills in C/C++, with a solid understanding of operating systems and computer architecture.

Responsibilities

• Adding optimized GPU compute kernels across Machine Learning, Image Processing, Linear Algebra and Computer Vision.
• Defining and implementing APIs in Metal Performance Shaders.
• Performing in-depth analysis, compiler and kernel level optimizations to ensure the best possible performance across hardware families.
• Partnering with Platform Architecture teams to define Apple GPU's compute hardware roadmap. Working with hardware team to analyze performance on future silicon
• Tune GPU-accelerated compute across products.

Skills

GPU kernel development

Distributed training

System-level programming

Leadership/mentorship

Education

BS/MS/PhD in Computer Science or related field

Tools

Metal

CUDA

CuTE

CuTile

Triton

LLVM

Job description

Cupertino, California, United States Software and Services

Apple’s Server ML Frameworks team in Intelligent System Experience works on enabling Apple Intelligence through high-performance, distributed inference of GenAI applications (such as LLMs) on Private Cloud Compute. You will get to work on custom-built server hardware that brings the power and security of Apple silicon to the data center. We are looking for a senior technical leader with strong technical background who is deeply passionate about building scalable, efficient, and production-grade solutions tailored for high-throughput GPU execution and to partner with software and hardware leaders to define Server Apple Intelligence roadmap.

Description

As a senior technical leader on Apple's Server ML Frameworks team, you will set the technical direction and software architecture for the frameworks that power Apple Intelligence at scale. You will partner with teams across Apple to propose, develop, and evaluate features in new Apple Intelligence models; collaborate closely with our hardware organizations to influence the design of next-generation silicon; and help define the compute and programming models for future server architectures. You will also grow and mentor a team of engineers, establish engineering best practices, and build the cross-functional relationships needed to deliver complex, multi-team initiatives.

Responsibilities

• Adding optimized GPU compute kernels across Machine Learning, Image Processing, Linear Algebra and Computer Vision.
• Defining and implementing APIs in Metal Performance Shaders.
• Performing in-depth analysis, compiler and kernel level optimizations to ensure the best possible performance across hardware families.
• Partnering with Platform Architecture teams to define Apple GPU's compute hardware roadmap. Working with hardware team to analyze performance on future silicon
• Tune GPU-accelerated compute across products.

Minimum Qualifications

• BS/MS/PhD in Computer Science or a related field, or equivalent practical experience
• 3–5 years of experience as a manager or technical leader
• Experience with GPU kernel development and optimization using compute programming models such as Metal, CUDA, or similar
• Experience with distributed training or inference techniques
• Strong system-level programming skills in C/C++, with a solid understanding of operating systems and computer architecture

Preferred Qualifications

• Track record in strategic leadership and decision-making
• Familiarity with Apple framework and API design patterns
• Experience with graph compilers such as CuTE, CuTile, Triton, OpenXLA or LLVM is a plus
• Good understanding of LLM and Diffusion based model architectures

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 $184,700 and $324,800, 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 still 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.