Aiml - ML Prototyping Engineer, Machine Learning Research

Apple Apple · Big Tech · Cupertino, CA +1 · Machine Learning and AI

This role is for an ML Prototyping Engineer within Apple's Machine Learning Research team. The primary focus is on bridging the gap between AI/ML research and product development by building prototypes. Responsibilities include training and fine-tuning models, experimenting with new techniques, creating end-to-end prototypes, and developing interactive experiences to showcase ML capabilities. The role requires strong ML fundamentals, experience with training/fine-tuning, and the ability to implement research papers. It spans the full stack of AI development, from model training to serving systems and interfaces.

What you'd actually do

  1. Train, fine-tune, and adapt foundation models for novel applications
  2. Experiment with emerging ML techniques and architectures
  3. Build end-to-end prototypes that demonstrate ML capabilities in context
  4. Create interactive experiences that communicate ideas across Apple and to the broader ML community
  5. Collaborate with researchers to translate emerging techniques into working systems

Skills

Required

  • Experience training or fine-tuning ML models (PyTorch, JAX, or MLX)
  • Strong understanding of ML fundamentals: model architectures, training dynamics, evaluation
  • Familiarity with current ML research landscape
  • Experience reading and implementing techniques from ML papers
  • Proficiency in Python

Nice to have

  • Swift
  • C++
  • Rust
  • Experience with LLMs: prompting, fine-tuning, RLHF, inference optimization
  • Experience reproducing results from ML papers
  • Familiarity with Apple platforms and frameworks (CoreML, Metal, SwiftUI)
  • Experience building native apps for iOS or macOS
  • Background in an R&D, research, or prototyping environment
  • Ability to work on ambiguous problems where the goal is learning, not shipping
  • Bias toward building—you'd rather make something to test an idea than debate it
  • Initiative to pursue ideas without waiting for direction
  • Eye for detail and craft in how you present work
  • Comfort communicating ML research to diverse technical audiences
  • Experience presenting at or attending ML conferences (NeurIPS, ICML, etc.)
  • Interest in AI education or open-source community building

What the JD emphasized

  • Track record of building complete, working systems rather than isolated components

Other signals

  • prototyping cutting-edge research
  • translating research into prototypes
  • building end-to-end prototypes