AIML - Sr Machine Learning Engineer, Data and ML Innovation

Do you want to play a part in the revolution in Foundation Models? Contribute to model hillclimbing for Apple Intelligence features that leverage Apple Foundation Models, and work with the people who built the intelligent products that helps millions of people get things done - just by asking or typing?\\n\\nThe vision for AIML FM Data organization is to improve Foundation Models by leveraging data and cutting-edge LLM techniques. As a Sr ML Engineering on the team, you will drive ML innovations, identify key opportunity areas and experiment with various techniques to improve model training and evaluation efficiency and performance.

As a Senior Machine Learning Engineer, you will join end-to-end development of large language models and agentic systems, from training pipelines to evaluation frameworks and production deployment. \nYou will work at the intersection of modeling, infrastructure, and product, helping push model quality through systematic experimentation and iteration. \nYou'll collaborate closely with research, infrastructure, and product teams to design robust training pipelines, build agent environments, and ship high-impact AI capabilities into real-world applications. \nThis role blends deep modeling expertise with strong engineering fundamentals and offers the opportunity to shape both the technical direction and the ML platform powering Apple products.

5+ years of hands on ML engineering experiences, with at least 1+ years working directly on large language models or generative AI.\nBachelor's, Master's, or PhD in Computer Science, Machine Learning, or a related technical field - or equivalent practical experience.\nHands-on experience with LLM training workflows, including one or more of: Pretraining or continued pretraining, Supervised fine-tuning (SFT), Preference optimization (e.g., RLHF, DPO, PPO).\nStrong software engineering fundamentals: debugging, testing, code reviews, and production reliability.\nDemonstrated publication records in relevant conferences (e.g., NeurIPS, ICML, ICLR, etc.).

Direct experience with agentic systems, including tool use, environment design, or reinforcement learning.\nExperience with building or operating training environments or simulators (gym-style, tool-based, or sandboxed environments).\nExperience with model hillclimbing workflows: systematic experimentation, ablations, dataset iteration, and continuous quality improvement.\nAbility to work across research and engineering boundaries, turning ideas into scalable systems.\nHave demonstrated creative and critical thinking with an innate drive to improve how things work. Have a high tolerance for ambiguity.