AIML - Machine Learning Engineer - Answers, Knowledge & Information (AKI)

The Answers, Knowledge and Information team is revolutionizing the way hundreds of millions of people access information on their devices, all while keeping user privacy at the forefront. As an Applied ML team, we're pushing the boundaries of Apple Intelligence, result ranking, and innovative search technologies, all while running a low latency production service. Our work fuels intuitive information experiences across some of Apple's most iconic products, including Siri, Spotlight, Safari, Messages, Lookup, and more. Join us in shaping the future of how the world connects with information!

We are seeking a Senior Machine Learning Engineer with a passion for applying machine learning and generative AI to create intelligent, personal search solutions for Apple products. You will research and develop novel approaches in query understanding, ranking, and ML modeling-building the technology that enables seamless and delightful search and assistant experiences across Apple products.

Description

This role offers the opportunity to research and develop personalized text, image, and multimodal retrieval and ranking models, while driving improvements and scalable evaluations that impact millions of Apple users and advance the broader AIML community.","responsibilities":"Research, design, implement, and evaluate algorithms and models to advance personalized query understanding, user intent resolution, retrieval, and ranking

Apply state-of-the-art NLP, CV, deep learning, and GenAI/LLM techniques to return grounded, contextually relevant responses

Analyze loss patterns in current search and assistant stacks, and develop insights, algorithms, and techniques to close quality gaps and improve key product metrics

Define and drive long-term technical vision, proposing clear roadmaps and setting measurable objectives for the team

Collaborate with cross-functional partners to define product requirements and prioritize ranking criteria that maximize user satisfaction

Establish robust evaluation metrics and continuously enhance model performance through innovative ML approaches

Preferred Qualifications

* Deep expertise in applying and evaluating deep learning models and large language models (LLMs)

Extensive experience building production-quality systems in search, recommendation, or personal assistant domains

Hands-on proficiency with modern ML frameworks such as PyTorch, JAX, TensorFlow, or XGBoost

Proven ability to rapidly prototype solutions and perform critical analysis

Strong background in personalization and data-driven decision making

Advanced degree (Master's or Ph.D.) in Computer Science, Statistics, or a related field, or equivalent industry experience

Minimum Qualifications

4+ years of experience in Machine Learning, NLP/CV, and applying these techniques at scale

Proficiency in software engineering with mainstream programming languages such as Python, Go, or C/C++

Strong communication and collaboration skills

Bachelor in Computer Science with relevant industry experience

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 .