AIML - Senior Machine Learning Platform Engineer, Machine Learning Platform and Technology jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 2 days ago

AIML - Senior Machine Learning Platform Engineer, Machine Learning Platform and Technology

Apple is seeking a Senior Machine Learning Platform Engineer to build the ML platform that supports various intelligent experiences across their products. In this role, you will create integrations between the ML platform and frameworks, design SDKs, and enhance the productivity of ML practitioners by simplifying infrastructure complexities.

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

Responsibilities

Build and scale core platform capabilities for model management and serving infrastructure
Create seamless integrations with the ML frameworks that Apple's teams depend on
Design and build systems that feel native to each framework while providing a unified experience across the platform
Build backend services
Design Python SDKs and APIs
Create integrations across ML tools and frameworks
Solve complex technical challenges that span multiple systems
Work closely with ML engineers to understand their workflows, identify pain points, and build platform features that multiply their productivity
Collaborate with teams building customer-facing ML features across iOS, macOS, and other Apple platforms
Collaborate with compute infrastructure teams and ML framework owners

Qualification

PythonML frameworks integrationBackend developmentCloud platformsModel serving systemsOpen-source contributionsMLOps practicesCross-functional collaborationCommunication skills

Required

Bachelor's degree in Computer Science, related field, or equivalent practical experience
10+ years of software engineering experience with strong backend development skills and platform engineering mindset
Deep proficiency in Python with proven experience designing SDKs, libraries, and APIs for technical users
Experience integrating with complex ML frameworks (PyTorch, TensorFlow, JAX, HuggingFace) and building production-grade backend services (REST/GraphQL APIs, microservices, databases)
Track record of building end-to-end workflows that span multiple systems and teams, navigating complex technical landscapes to deliver pragmatic solutions
Strong cross-functional collaboration and communication skills to understand diverse stakeholder needs, technical requirements, and articulate design decisions across ML engineering, infrastructure, and product teams
Experience with cloud platforms (AWS, GCP, Azure) and container orchestration (Kubernetes)

Preferred

Experience with model serving systems (vLLM, Ray Serve, TorchServe, TensorRT) or inference optimization
Contributions to open-source ML frameworks, tools, or libraries
Understanding of distributed training, model parallelism, and large-scale ML workflows
Familiarity with MLOps practices, model management, and experiment tracking systems

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