Machine Learning Engineer, Platform Architecture jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 3 months ago

Machine Learning Engineer, Platform Architecture

positionCupertino, California, United States
timeFull-time
remoteOnsite
seniorityMid, Senior Level
money$147K/yr - $272K/yr
date3+ years exp

Apple's Platform Architecture group is responsible for connecting hardware and software into one unified system. In this role, you will collaborate with engineers to design technologies, develop performance models for ML workloads on Apple silicon, and inform architecture decisions.

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

Responsibilities

Create optimized implementations of ML workloads on Apple silicon including Neural Engine, GPU, and CPU
Collaborate with IP and SoC architecture teams to develop performance models and simulations of future hardware
Conduct performance studies to inform and validate architecture decisions
Collaborate with system teams to create high-level performance models of emerging ML techniques and analyze system architecture trade-offs

Qualification

C/C++ programmingPython programmingML frameworksComputer architecture knowledgeML model optimizationPerformance modelingAlgorithm prototypingCompiler technologiesVerbal communicationWritten communication

Required

Bachelor's degree
Ability to program in C/C++ and/or Python
Knowledge of computer architecture fundamentals
Domain knowledge in at least one hardware IP: ML HW accelerators or processing units such as GPU, image/video, CPUs, or similar

Preferred

MS or PhD in EE/CE/CS or related field, or 3+ years of relevant experience
Experience with ML frameworks (e.g. PyTorch) and efficient implementations of machine learning algorithms
Experience in optimizing and deploying ML models and/or runtime frameworks in production inference/training environments
Experience in creating SoC or IP performance models/simulations
Verbal and written communication skills for collaborating with partner teams
Ability to prototype algorithms on CPU/GPU/Neural Engine, analyze performance metrics, and create high-level complexity models
Understanding of compiler frameworks/technologies

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