Senior Staff Machine Learning Engineer

What you'll do

• Lead a team of engineers to deliver efficient and effective Generative AI models for content summarization.
• Drive end-to-end R&D and engineering to create high-quality summaries for both on-device and cloud-based platforms.
• Develop state-of-the-art LLM-based generative, groundedness, and safety models to ensure accurate and concise outputs.
• Build sophisticated software frameworks for fast and cost-efficient LLM inference across on-device and server environments.
• Improve summarization quality for world knowledge questions and real-time web page highlights in Safari.
• Integrate the Apple ecosystem with internal LLM infrastructure to deliver high-quality user experiences.
• Define the product vision, strategy, and execution plans for generative AI features.

What we're looking for

• MS or Ph.D. in Computer Science, Machine Learning, information retrieval, data mining, or a related field.
• 8+ years of experience in leading engineering, applied research, or ML experiences in NLP and SOTA generative AI models.
• Proven record of consistent delivery of technology and products across the full Machine Learning life cycle.
• Strong background and experience in Machine Learning, NLP, and RAG.
• Experience in LLM post-training and advanced RL-based methods including RLHF/RLAIF, reward models, and policy optimization.
• Expertise in hallucination reduction methods and engineering high-availability, low-latency, and robust RL-based models.
• Exceptional verbal and written communication skills to lead teams and collaborate with partners.
• Strong engineering leadership, product vision, and business acumen to manage long-term strategy.