Senior / Staff Machine Learning Engineer, Apple ML Data Platform

Join a team at the forefront of ML infrastructure and generative AI, where data and model workflows come together to enable the next generation of intelligent experiences on Apple products and services

• Prototype and optimize GenAI models, including open-source models, for scalable production use
• Build a platform that enables teams to easily configure models, apply tuning strategies, perform quantization, and get models production-ready for scalable deployment
• Continuously improve platform capabilities to handle next-gen ML workloads, including foundation models and retrieval-augmented systems
• Use ML techniques to drive smarter data workflows - including synthetic data generation, automated labeling, active learning, and data curation
• Collaborate across research and engineering teams to accelerate experimentation
• Collaborate closely with teams across the stack to enable high-quality, end-to-end ML experiences
• Use and extend tools built on modern ML frameworks
• Optimize platform components for large-scale ML workloads across distributed systems
• Diagnose, fix, improve, and automate complex issues across the entire stack to ensure maximum uptime and performance

• Strong foundation in machine learning, with hands-on experience across the end-to-end ML workflow
• Familiarity with modern generative techniques (e.g. transformers, diffusion, retrieval-augmented generation)
• Proven experience building and delivering data and machine learning infrastructure in real-world production environments
• Familiarity with fine-tuning workflows, model optimization, and preparing models for scalable inference
• Familiarity with generative AI and its applications in accelerating and enhancing machine learning workflows
• Experience configuring, deploying and troubleshooting large scale production environments
• Experience in designing, building, and maintaining scalable, highly available systems that prioritize ease of use
• Extensive programming experience in Java, Python or Go
• Strong collaboration and communication (verbal and written) skills
• Comfortable navigating ambiguity and evolving technical landscapes, especially in fast-moving areas

• Comprehensive medical and dental coverage
• Retirement benefits
• Range of discounted products and free services
• Reimbursement for certain educational expenses
• Discretionary bonuses or commission payments
• Relocation assistance

Apple is a technology company that combines innovation, creativity, and collaboration to make products that enrich people's lives