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

What you'll do

• Design and implement experiments and methods to understand foundation models, and analyze the results.
• Communicate findings to Apple teams and collaborate with partner teams.
• Write papers for publication in top-tier conferences and journals.

What they ask for

• 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, including experience training them and applying them to complex real datasets, demonstrated through publications or code repositories.
• Experience designing experiments to understand foundation models; knowledge of Bayesian statistical methods and their use in scientific inference.

Nice to have

• Experience with large language models, including training them and using tools like vllm for inference.
• Proficiency implementing ML models and experiments in Python and Pytorch/Jax; familiarity with interpretability methods like activation patching/causal tracing.

Pay

$175,000 and $308,500 USD base salary

Tools

foundation models, vllm, Python, Pytorch, Jax

AI focus

Research how foundation models work and expand their capabilities in scientific domains; apply findings to practical problems.