AIML - Sr Machine Learning Engineer, Data and ML Innovation

Do you want to play a part in the next revolution in Foundation Models? Contribute to model hill climbing 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 the AI/ML FM Data organization is to improve Foundation Models by leveraging data from a variety of sources: crawl, license, vendor and internal crowd-sourcing. As a Sr ML Engineering on the team, you will drive ML innovations, identify key opportunity areas where data can play a crucial role and experiment with various data augmentation strategies to improve model training efficiency and performance..

We are looking for people with a track record in building models and model-driven products to affect user experiences. Join us, and impact hundreds of millions of customers across billions of their interactions with foundation model powered Apple Intelligence features, that are available on iPhone, iPad, HomePod, Mac, Watch, CarPlay, and tv across more than 30 languages.\n\n- Algorithm development: Define signals that are important in prompts, responses and CoT reasoning steps. These usually require a fine-tuned model for specific use cases.\n\n- Model evaluation: Understand the importance of a balanced eval-set. Ability to perform error analysis to figure out how to improve model capabilities.\n\n- Ablation experiments: Test your data augmentation strategies via ablation experiments. Comfortable debugging training errors, and tune hyper-parameters and data mixture to achieve desired outcome.\n\n- Data processing and data filtering: Ability to efficiently process and filter very large amounts of data, often times messy.

5+ years of hands on ML engineering experiences.\nMaster or PhDs in Computer Science, Electric Engineering or Mathematics.\nHave prior experience as an ML modeler/scientist/researcher. Knowledgeable in classic machine learning algorithms (SVM, Random Forest, Naive Bayes, KNN etc), as well as comfortable with more modern deep learning frameworks (PyTorch, Tensorflow, Jax). \nFamiliarity with multi-modal data and large models including image and video.\nPossess strong software engineering skills and mindset. Have a high bar for engineering code quality and scalability.

Hands on experiences with different phases in LLM model training, including LoRA, SFT, RLHF, reward modeling.\nA good communicator with clear and concise, active listening and empathy skills.\nAre self-motivated and curious. Strive to continually learn on the job.\nHave demonstrated creative and critical thinking with an innate drive to improve how things work. Have a high tolerance for ambiguity.