AIML - ML Prototyping Engineer, Machine Learning Research

Apple Inc.; Cupertino, Northern (CA, KY); Hybrid; USD 185,000 - 325,000; Full time

Benefits offered by this job

Comprehensive medical and dental cover

Retirement benefits

Employee stock programs

Tuition reimbursement

Relocation support

Job summary

Apple Inc. in Cupertino, California, seeks an AIML - ML Prototyping Engineer for ML research and prototype development.

You will train and fine-tune foundation models, build end-to-end prototypes, and craft interactive experiences that demonstrate ML capabilities within products and for the ML community. Ideal candidates combine ML research depth with practical systems-building skills, Python/Swift proficiency, and strong communication abilities to translate ideas into tangible Apple tech

Qualifications

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

Responsibilities

• Train, fine-tune, and adapt foundation models for novel applications
• Experiment with emerging ML techniques and architectures
• Build end-to-end prototypes that demonstrate ML capabilities in context
• Create interactive experiences that communicate ideas across Apple and to the broader ML community
• Collaborate with researchers to translate emerging techniques into working systems
• Iterate rapidly based on critique, treating prototypes as questions rather than answers

Skills

Python programming

Swift programming

ML fundamentals

Research communication

Tools

PyTorch

JAX

MLX

CoreML

Metal

SwiftUI

C++

Job description

AIML - ML Prototyping Engineer, Machine Learning Research

Cupertino, California, United States Machine Learning and AI

The Machine Learning Research Prototyping team sits at the intersection of cutting-edge research and real-world impact. We serve as the bridge between groundbreaking AI and ML discoveries and the innovative products that define tomorrow's user experiences. Our nimble, collaborative team transforms theoretical breakthroughs into tangible prototypes that help Apple understand where AI could go next, and help the broader ML community build on Apple research.We're a small team of engineers who blend ML expertise, product intuition, and design sensibility. We learn by building—our prototypes range from novel architectures to complete interactive experiences, whatever best illuminates a research idea. We work at the boundary between what models can do today and what Apple products and the ML community will need tomorrow.

Description

This role spans the full stack of AI development: from training and fine-tuning models, to building the systems that serve them, to crafting the interfaces that let people experience them. You'll iterate rapidly and communicate what you've learned.We value diverse experiences and unconventional paths—what matters is that you can build, learn, and communicate what you've discovered. We're looking for someone who brings something we don't already have, whether that's a novel technical specialty, an unusual background, or a way of thinking we haven't encountered. If you build things to understand them, we want to hear from you.

Responsibilities

• Train, fine-tune, and adapt foundation models for novel applications
• Experiment with emerging ML techniques and architectures
• Build end-to-end prototypes that demonstrate ML capabilities in context
• Create interactive experiences that communicate ideas across Apple and to the broader ML community
• Collaborate with researchers to translate emerging techniques into working systems
• Iterate rapidly based on critique, treating prototypes as questions rather than answers

Minimum Qualifications

• 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 and Swift; experience with C++ or Rust a plus
• Track record of building complete, working systems rather than isolated components

Preferred Qualifications

• 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
• Work you can share: models you have trained, systems you have built, ideas you have explored

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 $184,700 and $324,800, 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. Learn more about Apple Benefits
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.
Learn about accessibility in Apple’s workplace
Learn about reasonable accommodations for job applicants

Apple accepts applications to this posting on an ongoing basis.