Machine Learning Researcher, MLR

Requirements

• PhD, or equivalent practical experience, in Computer Science, or related technical field
• Demonstrated expertise in machine learning research.
• Ability to formulate a research problem, design, experiment, implement and communicate solutions.
• Publication record in relevant conferences (e.g., NeurIPS, ICML, ICLR, AISTATS, CVPR, ICCV, ECCV, ACL, EMNLP, etc).

Nice to Haves

• Hands-on experience working with deep learning toolkits such as JAX, PyTorch or MLX
• Proven industry experience
• Strong mathematical skills in differential calculus, probability, statistics.
• Strong coding skills, as exemplified by e.g. OSS contributions, and ability to maintain a coherent and evolving codebase.
• Ability to work as a team player in a diverse collaborative environment.
• You have proposed through previous publications impactful methods in areas of interest to the group, such as generative modeling (flow matching, diffusion, etc.), LLM/VLM training/fine-tuning/inference, neural network theory, or scaling laws.

What You'll Be Doing

• Define your research plan to advance our understanding of machine learning and execute it through implementation and experimentation, in collaboration with your colleagues.
• Provide technical mentorship and guidance, and prepare technical reports for publication and conference talks.
• Have the opportunity to collaborate with broader teams across Apple.