Research Scientist/Engineer - LLM Efficiency

Apple Inc.; Santa Clara, Northern (CA, KY); Hybrid; USD 185,000 - 325,000; Full time

Job summary

Apple Inc. in Santa Clara, CA, seeks research scientists and engineers to advance the efficiency of frontier models, including LLMs and diffusion models. You will design, train, and evaluate novel approaches, and collaborate with world-class ML researchers to push the state of the art.

You will publish findings, engage with academia, and ship research in collaboration with product teams across Apple. A strong prototype track record and deep learning expertise in PyTorch/JAX are expected.

Qualifications

• Significant experience building complete prototypes that achieve state-of-the-art performance in your areas of expertise.
• Publication record at 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.

Responsibilities

• Designing, training, and adapting frontier models to improve efficiency and performance.
• Designing and conducting empirical studies to better understand frontier models.
• Developing prototypes to demonstrate the effectiveness of the models and algorithms developed.
• Engaging with the academic community by publishing papers and presenting your work.
• Collaborating with product teams across Apple to ship research.

Skills

Python

Deep learning

Research prototyping

ML system design

Publication experience

Neural networks

Education

Bachelor's/Master's/PhD in Computer Science or related field

Tools

PyTorch

JAX

TensorFlow

Job description

Research Scientist/Engineer - LLM Efficiency

Santa Clara, California, United States Machine Learning and AI

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.

Description

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.

Responsibilities

• Designing, training, and adapting frontier models to improve efficiency and performance.
• Designing and conducting empirical studies to better understand frontier models.
• Developing prototypes to demonstrate the effectiveness of the models and algorithms developed.
• Engaging with the academic community by publishing papers and presenting your work.
• Collaborating with product teams across Apple to ship research.

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.

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $184,700 and $324,800, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses - including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.

Learn about accessibility in Apple’s workplace

Learn about reasonable accommodations for job applicants

Apple accepts applications to this posting on an ongoing basis.