AIML - Sr./Staff Machine Learning Engineer, Machine Learning Platform Technologies jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 2 months ago

AIML - Sr./Staff Machine Learning Engineer, Machine Learning Platform Technologies

Apple is a leading technology company known for its innovative products and services. They are seeking a Machine Learning Engineer with deep expertise in large-scale data and ML infrastructure to build and optimize data pipelines, enhancing the ML frameworks that underpin Apple’s knowledge and search systems.

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

Responsibilities

Tackle web-scale extraction and enrichment challenges, transforming raw web data into structured knowledge that powers Siri, Spotlight, Safari and other Apple experiences
Design and optimize pipelines that leverage large language models (LLMs), advanced NLP, and entity linking frameworks to identify, deduplicate, and contextualize information from the open web
Design and evolve the infrastructure that powers large-scale extraction and enrichment pipelines, enabling Apple to transform raw web data into high-quality knowledge used across various apple products
The data and insights you help generate will directly improve Apple’s foundation models, supporting better understanding, reasoning, and grounding
Working at the intersection of web data systems, large-scale ML infrastructure, and language modeling, you will shape how Apple builds and leverages knowledge at global scale

Qualification

Machine Learning workflowsLarge-scale data processingCloud-native environmentsGolangJavaScalaMicro-service architectureContainerized environmentsInterpersonal skillsTeamwork

Required

Bachelor's degree or higher in Computer Science or related technical field
Experience with Golang, Java or Scala
Background in computer science: algorithms, data structures, and distributed systems
Experience working in a cloud-native environment such as AWS
Experience working with large-scale data processing pipelines (Spark, Cassandra, etc.)
Experience with micro-service architecture in a containerized environment (Docker, Kubernetes, etc.)
Experience with machine learning workflows, including feature engineering, training, evaluation, deployment and serving

Preferred

Experience with training and fine-tuning large language models
Experience with optimizing ML training and serving performance, including GPU tuning, batch size optimization, and multi-node scheduling
Familiar with Nvidia TensorRT-LLM, vLLLM, Nvidia Triton Server etc
Excellent interpersonal skills, able to work independently as well as in a team

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