AIML - ML Engineer

Apple
San Francisco, US
On-site

Job Description

In this position, you will work with researchers across the Machine Learning Research group to build and explore speechLLMs and spoken dialog systems.

Work with researchers on the team to build high-performance and scalable software addressing novel ML research algorithms. Apply solid software engineering skills, leverage experience to deal with the unexpected, explore research software solutions and pave the way to future Machine Learning toolboxes. Be part of a small team dedicated to advancing ML algorithms and techniques - Is this you? If so we'd love to hear from you.

M.S. in Computer Science (or related fields) or equivalent experience

Expert in Python programming and tensor operations, including https://github.com/rougier/numpy-100

Expert in at least one ML framework (PyTorch preferred but tensorflow or jax are ok as well), including flash-attention, efficient kv-caching, DDP and FSDP.

Hands-on experience in establishing ML benchmarks (data, models, metrics)

Involvement with open-source projects and experience with collaborative software development

Excellent communication skills

Readiness to encounter unforeseen challenges and to solve them

Speech processing (including generative modeling of speech) experience is a plus

PhD in Computer Science (or related fields) or equivalent experience

Expert in Python programming and tensor operations, including https://github.com/rougier/numpy-100

Expert in at least one ML framework (PyTorch preferred but tensorflow or jax are ok as well), including flash-attention, efficient kv-caching, DDP and FSDP.

Hands-on experience in establishing ML benchmarks (data, models, metrics)

Active development in open-source projects and experience with collaborative software development

Experience with CUDA programming, and/or High-Performance Computing and/or distributed computing

Excellent communication skills

Readiness to encounter unforeseen challenges and to solve them

Past research experience

C++ programming is a plus

MLX experience is a plus

Skills & Requirements

Technical Skills

PythonPytorchTensorflowJaxCuda programmingHigh-performance computingDistributed computingC++MlxCommunicationProblem-solvingTeamworkSpeech processingGenerative modelingAi research

Employment Type

FULL TIME

Level

mid

Posted

3/23/2026

Apply Now

You will be redirected to Apple's application portal.

Sign in and we'll score your resume against this role.

Find Similar Jobs

Browse roles in the same category, level, and remote setup.