Apple · 20 hours ago
On-device ML Infrastructure Engineer (ML User Experience APIs & Integration)
Cupertino, CA
Full-time
Hybrid
Mid, Senior Level
$147K/yr - $272K/yrApple is at the forefront of an evolution where innovative AI meets the elegance of Apple silicon, and they are seeking an On-device ML Infrastructure Engineer to contribute to their Machine Learning team. The role involves developing APIs for ML engineers, integrating ML tools into repositories, and optimizing machine learning models for efficient performance on Apple platforms.
AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
Responsibilities
Develop APIs in Apple’s ML stack for ML engineers to efficiently import and implement their models
Integrate Apple’s ML tools into internal and external model repositories to demonstrate and stress-test model ingestion with peak efficiency and performance
Develop optimizations across the pipeline, including source-level transformations, and custom operations to improve inference efficiency
Onboard the latest ML models with peak performance, and use these examples to highlight and validate the authoring and runtime capabilities of Apple’s inference stack
Qualification
Required
Bachelors in Computer Sciences, Engineering, or related subject area
Highly proficient in Python programming, familiarity with C++ is required
Proficiency in at least one ML authoring framework, such as PyTorch, MLX, and JAX
Strong understanding of ML fundamentals, including common architectures such as Transformers
Hands-on experience with ML inference optimizations, such as quantization, pruning, KV caching, etc
Strong communication skills, including ability to connect with multi-functional audiences
Preferred
Experience with C++, Swift, and/or GPU programming paradigms
Familiarity with QAT and other compression and quantization techniques employing PyTorch workflows
Experience designing Python APIs and deploying production-grade Python packages
Experience with MLIR/LLVM or similar compiler toolchains
Familiarity with Hugging Face or other model repositories
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.
Founded in 1976
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