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
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
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.
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