On-Device ML Infrastructure Engineer (ML Insights and Forecasting) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 19 hours ago

On-Device ML Infrastructure Engineer (ML Insights and Forecasting)

positionSeattle, WA
timeFull-time
remoteOnsite
seniorityMid, Senior Level
money$140K/yr - $258K/yr
date4+ years exp

Apple is a leading technology company specializing in innovative machine learning solutions. The On-Device Machine Learning team is seeking an ML Infrastructure Engineer focused on ML Insights and Forecasting to enhance the development and performance of machine learning models on Apple devices.

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

Responsibilities

Explore the latest ML model architectures and prototype getting these running on device
Build infrastructure to enable at scale testing of new ML features
Analyze achieved performance vs roofline models on Apple’s hardware
Analyze telemetry data to understand how users are using ML on device
Identify gaps in today’s ML inference stack and work with XF teams to prioritize and address these
Collaborate extensively with ML and hardware teams across Apple

Qualification

Machine Learning technologiesOn-device ML frameworksPython programmingSystem performance analysisML model optimizationQuality practices knowledgeML fundamentalsCollaboration skillsCommunication skills

Required

Bachelors in Computer Science or relevant subject areas and and 4+ years of related experience, working with ML technologies
Experience with any ML authoring framework (PyTorch, TensorFlow, JAX, etc.), particularly on-device ML frameworks such as CoreML, TFLite or ExecuTorch
Strong programming and software design skills in Python
In depth knowledge of quality practices and fundamentals, including test planning, automation, and performance evaluation
Solid ML fundamentals including training regimes, evaluation and deployment/inference
Excellent collaboration and communication skills

Preferred

Masters or PhDs in Computer Science or relevant disciplines
Experience in system performance analysis and optimizing ML models for edge inference
Experience with standard ML concepts such as Transformers, CNNs or Stable Diffusion a strong plus

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

twittertwittertwitter
Glassdoor4.2
company-logo
Apple is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.
dateFounded in 1976
location
Cupertino, California, USA
people-size10001+ employees
web-link
https://www.apple.com

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

Recent News

Venrock
Venrock Portfolio
2025-12-01
IndiaTimes
Apple TV+ pulls Jessica Chastain’s "The Savant" just days before premiere
2025-09-25
Mac Daily News
Jessica Chastain: I respect Apple’s decision to pause the release of ‘The Savant’ or something
2025-09-25
Company data provided by crunchbase