AIML - Machine Learning Research Scientist, Data and ML Innovation

AIML - Machine Learning Research Scientist, Data and ML Innovation

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.

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 $139,500 and $258,100, 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.