Machine Learning for GenAI, AiDP

Apple's Generative AI team is looking for a creative and driven ML Engineer to help shape the future of AI-powered experiences across the enterprise. In this role, you will work on cutting-edge systems that enable large-scale training, inference, and deployment of generative models. If you are passionate about building production-grade ML systems, solving hard research-backed engineering problems, and contributing to innovations that impact millions of users, we want to hear from you.

Description You are a creative problem solver with strong ML and engineering skills who thrives in a fast-paced environment, working across teams and organizations. You enjoy learning new technologies and have a deep interest in ML/GenAI models and systems. You take ownership of your work, communicate scope clearly, and are motivated by the impact your models and systems will have on real-world users. Most importantly, you care about building responsibly and sharing your learnings with the broader ML community. The main responsibilities for this position include:

Responsibilities

• Develop ML/GenAI models and pipelines to solve for business KPIs at scale
• Innovate training pipelines using cutting-edge distributed systems and hardware-aware optimization
• Partner with cross-functional teams to translate groundbreaking research into user-centric products
• Tackle unique challenges in privacy-preserving generation, efficient inference, and multimodal integration
• Deliver production-grade models that meet Apple's rigorous standards for quality, performance, and scalability

Minimum Qualifications

• Bachelor of Science in Computer Science, Machine Learning, or a related quantitative field or equivalent experience
• 2+ years of hands-on experience in machine learning engineering
• 1+ years focused on generative AI or LLM technologies, or Agentic workflows
• Solid experience in Python
• Experienced building ML frameworks (PyTorch, JAX) for training, fine-tuning, and deploying generative models at scale
• Experience building enterprise-grade ML pipelines (data prep, distributed training, optimization, monitoring) in cloud environments (AWS, Google Cloud Platform, Azure) or on-prem infrastructure
• Deep understanding of transformer architectures, prompt engineering, retrieval-augmented generation (RAG), and LLM evaluation methodologies
• Experience optimizing models for latency, cost, and scalability (quantization, distillation, hardware-aware ML)

Preferred Qualifications

• Contributions to major open-source ML frameworks or research communities
• MS or PhD in Computer Science, Machine Learning, or a related quantitative field
• Solid grasp of NLP techniques, multimodal AI (text, image, code), and agent workflows.
• Experience with LLM Agentic workflows and framework (Langchain, LangGraph, LlamaIndex, CrewAI etc.)
• Knowledge in compiler/runtime optimizations for machine learning workloads

Pay & Benefits At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $181,100 and $318,400, and your base pay will depend on your skills, qualifications, experience, and location.

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. Learn more about Apple Benefits.

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

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. Learn more about your EEO rights as an applicant .