AIML - Sr Machine Learning Research Scientist, Data and ML Innovation

Apple Inc.; Seattle, Northern (WA, KY); Hybrid; USD 175,000 - 309,000; Full time

Benefits offered by this job

Medical and dental coverage

Stock programs

Tuition reimbursement

Relocation assistance

Discounted Apple products

Job summary

Apple Inc. seeks a senior AIML Research Scientist in Seattle to advance fundamental research on foundation models for scientific domains and collaborate with partner teams across Apple. You will plan, run, and publish experiments, translate findings into product directions, and contribute to papers for publication.

You will join a small team to define projects, develop methods, run experiments, and analyze results, with opportunities to engage in cross-team collaboration and publication.

Qualifications

• PhD in a relevant hard science with strong ML foundation.
• At least 3 publications in top ML/AI venues.
• Deep knowledge of foundation models and applications.
• Experience designing experiments to study foundation models.
• Knowledge of Bayesian statistics for scientific inference.

Responsibilities

• Design robust experiments to understand foundation models.
• Implement methods in experiment pipelines.
• Analyze and interpret experimental results.
• Communicate results across teams and publish papers.
• Write papers for top conferences/journals.

Skills

Publications

Foundation models

Experiment design

Bayesian methods

Python & PyTorch

Education

PhD (CSE/Stats/Physics/Chemistry/EE/OR)

Tools

Pytorch

Jax

vLLM

Job description

AIML - Sr Machine Learning Research Scientist, Data and ML Innovation

Seattle, Washington, United States Machine Learning and AI

Would you like to join a team curious about understanding how foundation models work and to expand their capabilities in scientific domains? We perform and publish novel research and apply our findings to drive product directions. If you like designing clever approaches to understanding complex phenomena and using that knowledge to solve practical problems then this is the position for you.

Description

In this research scientist role you will join a small team of researchers performing fundamental research investigating foundation models for scientific domains. You will be involved in all aspects of performing research including project definition, method development, and experimental design. You will also run your own experiments, then analyze and interpret the results. You will take an active role in writing papers for publication and also using our findings to solve applied problems. You will also have the opportunity to collaborate with partner teams across Apple.

Responsibilities

• Design robust experiments and methods to understand what foundation models do under the hood.
• Implement your methods and designs in experiment pipelines.
• Analyze and interpret your experimental results.
• Communicate results to teams across Apple.
• Write papers for publication in top-tier conferences/journals.

Minimum Qualifications

• PhD in computer science, statistics, physics, chemistry, electrical engineering, or operations research. Other hard sciences may also be considered.
• 3 publications in top-tier machine learning, statistics, or natural language processing venues.
• Deep knowledge of foundation models and experience training them and applying them to real, complex datasets as demonstrated through publications or code repositories.
• Experience designing experiments to understand how foundation models work.
• Knowledge of Bayesian statistical methods and how they are used for scientific inference.

Preferred Qualifications

• Experience with large language models and – both training them and using tools like vllm for inference.
• Proficient implementing ML models and experiments in Python and Pytorch/Jax.
• Familiarity with interpretability methods like activation patching/causal tracing.
• Demonstrate the ability to refine ambiguous research ideas to construct a coherent and logically sound story.

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 $175,000 and $308,500, 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.