Apple · 3 weeks ago
On-Device ML Compiler Engineer (Model Compilation)
Cupertino, CA
Full-time
Hybrid
Mid Level
$147K/yr - $272K/yr
3+ years expApple is at the forefront of an evolution where innovative AI meets the elegance of Apple silicon. The On-Device Machine Learning team is seeking an ML Infrastructure Engineer to focus on model compilation and enhance the execution of machine learning models across various Apple devices.
AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
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
Own core pieces of the compiler stack enabling heterogeneous compute across Apple devices. We target execution of ML models across the Apple ecosystem from resource-constrained devices like Apple Watch, to the high-end Macs with Ultra SoCs
Work closely with hardware, software, and performance teams across the company to accelerate and optimize execution by taking advantage of the latest features in the hardware, OS, and drivers
Qualification
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
Apple is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.
10001+ employees
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 CompanyTotal Funding
$5.67BKey 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
Tim Cook
CEO
Craig Federighi
SVP, Software Engineering
Recent News
Venrock
2025-12-01
2025-09-25
Mac Daily News
2025-09-25
Company data provided by crunchbase