On-Device ML Compiler Engineer (Model Compilation) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 2 months ago

On-Device ML Compiler Engineer (Model Compilation)

Apple is at the forefront of an evolution where innovative AI meets the elegance of Apple silicon. The On-Device Machine Learning team transforms groundbreaking research into practical applications, enabling billions of Apple devices to run powerful AI models locally, privately, and efficiently. They are seeking an ML Infrastructure Engineer focused on model compilation to build essential infrastructure for machine learning at scale on Apple devices.

AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Inspire changes in our MLIR-based compiler in order to target improved runtime performance by demonstrating the capabilities of the hardware
Propose upstream changes in MLIR to better support new features and workflows in the hardware that lead to more optimal execution performance across all types of devices and device clusters

Qualification

MLIR-based compilersML model architecturesC++PyTorchSwiftSystem software engineeringCompiler optimizationsGPU programmingCPU programmingNeural Engine programmingWriting kernels

Required

3-5 years working on MLIR-based compilers
Familiarity with common ML model architectures, execution schemes, and operations
Familiarity with C++
Familiarity with PyTorch or related training frameworks

Preferred

Familiarity with Swift
Familiarity with programming paradigms for the GPU, CPU, and Neural Engine
Familiarity with writing kernels for ML model execution

Benefits

Comprehensive medical and dental coverage
Retirement benefits
A range of discounted products and free services
Reimbursement for certain educational expenses — including tuition
Discretionary bonuses or commission payments
Relocation

Company

Apple is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.

H1B Sponsorship

Apple has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (6998)
2024 (3766)
2023 (3939)
2022 (4822)
2021 (4060)
2020 (3656)

Funding

Current Stage
Public Company
Total Funding
$5.67B
Key Investors
Berkshire HathawayMicrosoftSequoia Capital
2025-05-05Post Ipo Debt· $4.5B
2025-01-16Post Ipo Debt· $0.31M
2021-04-30Post Ipo Equity

Leadership Team

leader-logo
Tim Cook
CEO
leader-logo
Craig Federighi
SVP, Software Engineering
Company data provided by crunchbase