AIML Prototyping Engineer: Build ML Systems & Prototypes

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

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