Apple · 1 month ago
AIML - Sr. Machine Learning Infrastructure Engineer, Evaluation
San Francisco, California, United States
Full-time
Onsite
Senior Level, Lead/Staff
$181K/yr - $318K/yr
10+ years expApple is a leading technology company known for its innovative products and services. The AI/ML Evaluation team is seeking a Staff Engineer to lead the creation of tooling for synthetic data generation, ensuring the robustness and fairness of AI models across Apple products. This role involves architecting systems for generating diverse datasets to challenge AI models and enhance evaluation methodologies.
AppsArtificial Intelligence (AI)BroadcastingConsumer ElectronicsDigital EntertainmentMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
Responsibilities
Architect a highly scalable, multi-modal platforms for generating synthetic data using the latest generative models
Design and build robust, high-performance microservices in Golang to serve as the backbone of the data generation platform
Operate and scale distributed compute infrastructure, including large Apple internal job scheduling environments and dedicated GPU clusters
Develop resilient data pipelines for the curation, processing, and management of massive synthetic datasets
Define the technical strategy for integrating synthetic data into our core AI evaluation and testing workflows
Collaborate closely with research scientists and ML engineers to develop novel generative models for evaluation purposes
Optimize the computational efficiency and scheduling of data generation workloads, with a deep focus on maximizing GPU utilization
Mentor engineers across the organization on best practices for building and scaling distributed systems for generative AI
Qualification
Required
10+ years of professional software engineering experience building and operating large-scale, high-performance distributed systems
Strong programming skills in Go and Python, with proven experience building production services
Deep theoretical and practical knowledge of distributed systems principles (e.g., consensus, consistency, scalability)
Hands-on expertise with container orchestration and infrastructure-as-code in a production environment
Experience designing and operating infrastructure for machine learning workloads on GPU compute
BS in Computer Science or equivalent work experience
Preferred
Direct experience architecting systems for training or running large-scale generative models (e.g., Diffusion Models, GANs, LLMs)
Familiarity with using synthetic data for model testing, validation, robustness checks, or fairness evaluation
Architectural ownership of a large-scale ML platform or microservice-based system in a production environment
Strategic leadership in defining technical roadmaps and influencing cross-functional teams in an ambiguous, fast-paced domain
MS or PhD in Computer Science or a related field
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
Apple is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.
10001+ employees
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 CompanyTotal Funding
$5.67BKey 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
Tim Cook
CEO
Craig Federighi
SVP, Software Engineering
Recent News
Venrock
2025-12-01
2025-09-25
Mac Daily News
2025-09-25
Company data provided by crunchbase