Research Scientist - LLM Efficiency

We're seeking research scientists and engineers to create breakthrough innovations in machine learning. We are particularly interested in efficiency of frontier models, including LLMs and diffusion models. You will work in an organization of world-class machine learning researchers and engineers. Our work powers cutting-edge technologies across the Apple ecosystem and is published in the most selective scientific journals and conferences.

We are a team of best-in-the-world research scientists and engineers with deep experience in machine learning. We work on exciting new technologies that bring joy to millions of people. In our daily work, the team stays innovative, productive, and fun by sharing several key values:

• Passion for the mission: We're here to make something extraordinary. We seek out whatever work is right and strive for the best possible results.
• Modesty: Finding the right answer is more important than being right. We search for solutions as a team and value clear-eyed feedback..
• Lean habits: You can't grow without limits. Time constraints and big goals encourage us to sharpen our focus and learn to make phenomenal decisions.

Minimum Qualifications

• Significant experience building complete prototypes that achieve state-of-the-art performance in your areas of expertise.
• A publication record at relevant conferences, such as NeurIPS, ICML, ICLR, and COLM.
• Comfort working across the entire ML research stack, from low-level systems programming to high-level algorithm development.
• Solid software engineering skills and experience working with complex systems. Fluency in Python.
• Expertise in ML algorithms and best practices for working with deep learning systems.
• Proficiency with ML modeling frameworks such as PyTorch and JAX.

Preferred Qualifications

• A bachelor's, master's, or PhD degree in Computer Science, Electrical Engineering, Mechanical Engineering, or a related field, or equivalent academic or professional experience.