AIML Engineer: Build Scalable ML Pipelines & GPU Systems

Apple Inc.; Cupertino (CA); On-site; USD 184,700 - 324,800; Full time

Benefits offered by this job

Medical & dental

Retirement benefits

Discounted products

Employee stock programs

Job summary

Apple Inc. in Cupertino, California, is seeking an AIML - Machine Learning Engineer for MLX, MLR to join the Machine Learning Research team. You will collaborate with researchers and software engineers to build scalable training and research pipelines on Apple silicon.

The role focuses on high-performance software for novel ML algorithms, helping advance ML toolboxes and contribute to future research software in a dynamic team.

Qualifications

• 2 years in C++ and Python numerical computing
• 2 years low-level programming
• 2 years in GPU programming
• Experience with open-source projects and collaborative software development.
• Experience with Machine Learning algorithms.
• B.S., M.S. or PhD in Computer Science with a strong mathematical background

Responsibilities

• Work with researchers and software engineers on the MLX team to build high performance and scalable software addressing novel ML research algorithms.
• Pave the way to future Machine Learning toolboxes and contribute to scalable training and research pipelines.

Skills

C++ & Python numerics

Low-level programming

GPU programming

Open-source

ML algorithms

C++ & Python programming

Swift programming

Problem solving

Research experience

Education

B.S./M.S./PhD in CS

Tools

Swift programming

Job description

Apple Inc. in Cupertino, California, is seeking an AIML - Machine Learning Engineer for MLX, MLR to join the Machine Learning Research team. You will collaborate with researchers and software engineers to build scalable training and research pipelines on Apple silicon.

The role focuses on high-performance software for novel ML algorithms, helping advance ML toolboxes and contribute to future research software in a dynamic team.