Senior Machine Learning Engineer

New York City, New York, United States Software and Services

At Apple, we work every day to create products that enrich people's lives. Our Ad Platforms group makes it possible for people around the world to easily access informative and imaginative content on their devices while helping publishers and developers promote and monetize their work. Today, our technology and services power advertising in Search Ads, App Store, and Apple News. Our platforms are highly-performant, deployed at scale, and setting new standards for enabling effective advertising while protecting user privacy. The Machine Learning Platform team's mission is to empower Ad Platforms teams to build and scale the innovative ML systems that deliver highly optimized advertising content to consumers. Are you a results-oriented and versatile engineer who can excel in an Agile environment? You will work closely with engineers and data scientists to design, develop, and build world-class platform capabilities that will enable Ad Platforms teams to improve and scale our ML features, models, and applications.

The ML Platform team is responsible for bringing numerous features to advertisers and consumers while simultaneously supporting scalable modeling and continuous experimentation by all Ad Platforms teams. As a key contributor to this team, you will design and develop secure and scalable back-end systems. You will enjoy building high-performing, elegant systems from the ground up, in close partnerships with various teams. You will also possess keen judgment in selecting technologies and building the right solution for the interesting challenges we get to tackle here. You will have the opportunity to define and refine architectures to meet the unique ad network challenges we must solve. You will play a meaningful role building machine learning products which deliver on Apple's privacy commitments and change the way advertising works with data. Join us and contribute to a culture that emphasizes reliability, simplicity, and scalability. You will join a team of world-class machine learning engineers hungry to apply leading-edge technologies to deliver extraordinary experiences to our customers. We are one team, nurturing each other's growth and supporting each other in delivering for our customers!

Minimum Qualifications

• Proven track record of designing and scaling robust ML infrastructure and frameworks that support both training and inference across teams and orgs.
• Experience of model quantization, tensor parallelism, and inference optimizations (e.g. ONNX Runtime, TensorRT, vLLM). Actively led evaluation and adoption of such technologies.
• Recognized as a technical leader and mentor, supports the growth of engineers through code/design reviews, working groups, and internal knowledge sharing.
• Experience building machine learning models using frameworks like PyTorch, TensorFlow. Provides technical guidance and mentorship on best practices.
• Prior experience in advertising industry, federated learning and privacy-preserving ML techniques.
• Led development of foundational AI/ML platforms and tooling including Feature Stores, Vector DB to accelerate team productivity and model lifecycle management.
• Experience working on distributed systems (e.g. Ray, Spark, Kubernetes).
• Experience performance tuning & trouble-shooting.

Preferred Qualifications

• Pride in building tools to automate routine tasks, organized & detailed.
• Passionate about developer experience builds abstractions, automation tools, and reusable components to streamline ML workflows and reduce operational burden.
• Ability to communicate effectively, both written and verbal, with technical and non-technical multi-functional teams.
• Results oriented with a desire to work in a fast-paced and collaborative work environment.
• PhD/MS/BS in computer science or related field with 8+ years of experience in machine learning and strong software engineering skills.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple's discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple's Employee Stock Purchase Plan. You'll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, veteran status, or other legally protected characteristics.