AIML - Staff ML System Engineer, ML Platform Technologies (MLPT) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 1 day ago

AIML - Staff ML System Engineer, ML Platform Technologies (MLPT)

Apple is seeking a machine learning engineer to build the machine learning platform for next-generation Apple Intelligence products and services. The role involves creating platforms and tools for scalable ML workloads and participating in open source projects and top ML conferences.

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

Responsibilities

Create platforms and tools to enable performant, scalable ML workload for Apple’s AI-driven experiences
Design and develop components for our centralized, scalable ML platform
Push the limits of existing solutions for large-scale ML workloads
Develop novel techniques to circumvent the limitations of these solutions
Deploy your techniques on high-impact tasks from our partners across the company building new Apple Intelligence products and services
Encourage publishing novel work at top ML conferences and releasing contributions as open source

Qualification

Deep learning infrastructureDistributed systemsPython programmingDeep learning frameworksCloud platformsModel training optimizationData structuresSoftware design principlesAlgorithmsGo programmingKubernetesKafkaPrometheusFoundation model architecturesPublication record

Required

Experience building large-scale deep learning infrastructure or platforms for distributed model training
Experience with large-scale AI training infra components, such as accelerators, network fabrics, CUDA, NCCL, RDMA
Strong programming skills in Python or Go
Understanding of data structures, software design principles, and algorithms
Experience building large-scale distributed systems with tools such as Kubernetes, Kafka, Prometheus, etc
Experience with deep learning frameworks, such as PyTorch, or JAX
With minimum of 7+ years of industry experience
Bachelors degree in the area of Computer Science or equivalent, or a related domain

Preferred

Experience working with public cloud vendors such as AWS, GCP, Azure
Experience developing model parallel and data parallel training solutions and other training optimizations
Familiarity with recent developments in foundation model architectures for language and multimodal
Publication record at ML conferences such as MLSys, NeurIPS, etc
Advance degree in the area of Computer Science or equivalent, or a related

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