Finding the best job has never been easier
Apple AIML - ML Engineer United States, California, Cupertino 625115317
09.07.2025
Share
- 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
- 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
Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.
These jobs might be a good fit
