Drimva

Apple

AIML - Machine Learning Engineer for MLX, MLR

$185K–$325K

Cupertino, California · Full-time · Posted Mar 9, 2026

Experience
At least 2 years in C++ and Python numerical computing, low-level programming, and GPU programming; preferred: 3+ years in C++ and Python programming and 3+ Expert in GPU programming.
Education
B.S., M.S. or PhD in Computer Science with a strong mathematical background.

About the job

Build scalable, high-performance software and distributed training and research pipelines for ML research algorithms on Apple silicon as part of the MLX team.

What you'll do

  • Collaborate with researchers and software engineers to build scalable software for novel machine learning research algorithms.
  • Develop distributed training and research pipelines.
  • Advance machine learning algorithms and techniques on Apple silicon.

What they ask for

  • 2 years in C++ and Python numerical computing.
  • 2 years low-level programming.
  • 2 years in GPU programming.
  • Experience with open-source projects, collaborative software development, and machine learning algorithms.

Nice to have

  • Experience with Swift programming.
  • Past research experience.

Pay

$184,700 and $324,800 (base salary)

Tools

C++, Python, GPU, Swift, MLX, Apple silicon

AI focus

Build software and research pipelines for machine learning algorithms and techniques.

Model-written summary · the original posting is authoritative

First seen Oct 15, 2026 · Last verified Nov 5, 2026 (UTC)