Aiml - Machine Learning Research Scientist, Data and ML Innovation

Apple Apple · Big Tech · Seattle, WA · Machine Learning and AI

Research Scientist role focused on fundamental research of foundation models for scientific domains, involving project definition, method development, experimental design, analysis, interpretation, publication, and applied problem-solving. Collaborates with internal teams.

What you'd actually do

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

Skills

Required

  • PhD in computer science, statistics, physics, chemistry, electrical engineering, or operations research.
  • Deep knowledge of foundation models
  • Experience training foundation models
  • Experience applying foundation models to real, complex datasets
  • Experience designing experiments to understand how foundation models work
  • Knowledge of Bayesian statistical methods

Nice to have

  • Experience with large language models
  • Experience training LLMs
  • Experience 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
  • Ability to refine ambiguous research ideas to construct a coherent and logically sound story

What the JD emphasized

  • 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.

Other signals

  • novel research
  • foundation models
  • scientific domains
  • understanding complex phenomena
  • solve practical problems