AIML – Machine Learning Engineer, Foundation Models – Apple

Apply Now

Job Description



  • MS or PhD in Computer Science, Machine Learning, or related technical field
  • Expert-level programming skills in Python
  • Proficiency in machine learning frameworks such as Jax, PyTorch, TensorFlow
  • Strong background in: Distributed training, Model optimization, and Machine learning infrastructure
  • Experience with large-scale model training and deployment
  • Familiarity with: Kubernetes, Docker, Cloud platforms (AWS, GCP, Azure)
  • Distributed computing frameworks


  • Experience with foundation models and large language models
  • Background in multi-modal AI systems
  • Demonstrated ability to transform research prototypes into production systems
  • Published research or significant contributions to open-source ML projects
  • Understanding of on-device machine learning techniques