AIML - Machine Learning Researcher, Foundation Models
Core
Researching and developing frontier foundation models, including vision-language modeling and image/video generation, optimized for Apple silicon and integrated into the OS.
Role type
Senior IC machine learning researcher (foundation models)
Builds
Production foundation models and evaluation suites for visual understanding and generation
Domain
Consumer technology, deep learning, multimodal AI
Deliverable
production ML models
Required skills
deep learning research, Python programming, JAX/PyTorch/Tensorflow, vision-language modeling, image/video generation, reinforcement learning, on-policy distillation
Preferred skills
training/adapting vision-language models, post-training/mid-training LLMs, multimodal models
Technologies
JAX, PyTorch, Tensorflow
Responsibilities
Develop mid-training approaches bridging general capability and task-specific performance; solve reward modeling problems to resist reward hacking; align models across open-ended creative tasks and precise workflows; build data pipelines and evaluation suites for visual understanding at scale
Seniority
Senior, hands-on IC
