Pre-silicon ML and Compute Framework Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 9 hours ago

Pre-silicon ML and Compute Framework Engineer

Apple’s Graphics, Games, and Machine Learning team provides the compute and graphics software foundation across all of our innovative products. The Pre-silicon Compute Frameworks team is seeking a senior engineer to influence and lead new features in Metal and Metal compute frameworks through our next-generation GPU hardware roadmap.

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

Responsibilities

Develop and deliver software representing Apple’s roadmap for compute and ML technologies during hardware development
Design APIs that enable developers to make full use of new hardware features in the simplest way possible
Partner with cross-functional teams to ensure our software and hardware roadmap includes the state-of-the-art features and performance
Contribute to the bring-up and verification of new chips and products using our new GPUs

Qualification

GPU compute APIsMachine learning fundamentalsC/C++ programmingOperating Systems knowledgeComputer Architecture knowledgeML frameworksAPI designCollaboration skills

Required

BS/MS/PhD or equivalent in Computer Science or related field with 5 years experience
Strong grasp and practical experience with at least one GPU compute APIs (Metal, CUDA, OpenCL, etc.)
Good understanding of machine learning fundamentals and familiarity with ML frameworks (PyTorch, JAX, CoreML)
Knowledge of Operating Systems, Computer Architecture, and CPU/GPU programming
Strong background in C/C++ programming

Preferred

Experience collaborating in large cross functional projects
Familiarity with Apple framework and API design patterns
Hands on experience with ML frameworks (PyTorch, JAX, CoreML)

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 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 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
Company data provided by crunchbase