AIML - Machine Learning Engineer, MLR

Paris, Paris Region, France
Machine Learning and AI


Role Number:200517435
As part of Apple’s Machine Learning Research (MLR) organization, we carry out open scientific research to build the technologies that will power future products at Apple. The MLR team does world-class research and development across a wide range of domains including understanding and improving core methods in ML, addressing bias and fairness in algorithms, privacy and federated learning, embodied and interactive ML systems, computer vision, generative and large language models, and more. The techniques and tools we create and publish will impact ML solutions across Apple, which in turn power most of the features we deliver to billions of consumers worldwide. A software engineer in this position will work with researchers across MLR to build scalable, distributed training and research pipelines as well as implement research ideas into algorithms and conduct experiments. Candidates will need strong interpersonal skills to work with a diverse set of partners and customers to determine the right solutions for the problems. They will learn industry standard and custom tools/infrastructure to build these solutions. This work will directly benefit the researchers across our team and all of Apple.

Key Qualifications

  • Expert in Python programming and experience with PyTorch, JAX.
  • Experience with distributed computing
  • Experience maintaining an open-source project
  • Excellent interpersonal skills
  • Experience and passion to circumvent unexpected roadblocks
  • Experience in the fast-paced environment of ML research, either through a degree (e.g. PhD) and demonstrated impact on publications, or through relevant product experience.
  • Experience with CUDA programming, and/or High-Performance Computing is a plus


Develop and maintain machine learning infrastructure for our ongoing efforts. Most importantly, we need to support a wide variety of datasets, develop a scalable and flexible training infrastructure and support new features as they increase in priority during the course of research. Accelerate our research efforts through experimentation and comprehensive evaluation of model performance. Become our resident expert on the machine learning infrastructure at Apple. Investigate and understand various machine learning infrastructures that we may wish to em for MLR. Advise teammates within MLR about new technologies both within and external to Apple for performing machine learning research.

Education & Experience

M.S. or Ph.D. in Computer Science (or related fields), or B.S. with equivalent experience.

Additional Requirements