Aiml – Staff Machine Learning Engineer – Siri and Information Intelligence

Aiml – Staff Machine Learning Engineer – Siri and Information Intelligence

Requirements

• Master’s in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• 10 years of work experience in machine learning, deep learning or related field
• Experience working with Deep learning or LLM model development for various NLP tasks and RAG applications including prompt engineering, training data collection and generation, model fine-tuning and model evaluation
• Experience working with Python and at least one of the deep learning frameworks such as TensorFlow, PyTorch, or JAX.

Responsibilities

• Conduct research and development on state-of-the-art deep learning and large language models for various tasks and applications in Apple’s AI-powered products
• Developing, fine-tuning, and evaluating domain-specific Large Language Models for various NLP tasks including summarization, question answering, search relevance/ranking, entity linking and query understanding problems
• Conducting applied research to transfer the cutting edge research in generative AI to production ready technologies
• Understanding product requirements, translate them into modeling tasks and engineering tasks
• Stay up to date with the latest advancements and research in deep learning and large language models

Preferred Qualifications

• PhD in Computer Science, Artificial Intelligence, Machine Learning, or a related field.
• At least 2 years of experience in various state-of-the-art techniques related to LLM fine-tuning in 1 or more of the following areas: Supervised Fine-tuning (SFT) with Rejection Sampling, Preference-based fine-tuning techniques (e.g RLHF, Reward model, DPO, PPO, GRPO etc.), Parameter efficient fine-tuning techniques (e.g LoRA), Hallucination reduction and factual accuracy improvements, Designing and implementing safety guardrails
• At least 10 years of experience leading complex cross-functional research projects and influencing research direction
• At least 10 years of experience with large-scale model training, optimization, and deployment
• One or more scientific publications in various conferences and journals
• Consistent track record of researching, inventing and/or shipping advanced machine learning models
• Outstanding communication and interpersonal skills with ability to work with cross-functional teams.