Expoint – all jobs in one place
המקום בו המומחים והחברות הטובות ביותר נפגשים
Limitless High-tech career opportunities - Expoint

Apple AIML - ML Engineer 
United States, California, Cupertino 
625115317

09.07.2025
  • 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.