ML Software Research Engineer

Apple Apple · Big Tech · Pittsburgh, PA +1 · Machine Learning and AI

ML Software Research Engineer at Apple's MIND team, focusing on building end-to-end ML training and evaluation pipelines, developing production-quality runtime inference infrastructure, and optimizing on-device models for Apple features. The role involves collaboration with researchers and engineers to advance both research and production efforts, with an emphasis on shipping ML-based products.

What you'd actually do

  1. Build end-to-end ML model training and evaluation pipeline
  2. Develop ML production software
  3. The ability to build end-to-end demos for ML solutions
  4. Represent the team in Cross-functional discussions

Skills

Required

  • BS or MS + 3-5 years of ML SW Engineering or PhD in Computer Science, Computer Engineering or Relevant Fields
  • Strong foundation in Machine Learning (classical ML models, DNN, and multimodal Foundation Models)
  • Experience in building model training/eval pipelines in Python/Pytorch
  • Experience in prototyping and developing software applications
  • Strong communication and presentation Skills
  • Ability to work in a collaborative environment

Nice to have

  • Proficient programming skills in Python and one of the deep learning toolkits such as JAX, PyTorch, or Tensorflow
  • Experience in On-Device ML model deployment and on-device optimization
  • Experience handling multimodal data including text, images, audio, and other sensors
  • Experience in developing production software
  • Proficient in Swift, Objective C, C++, or Go

What the JD emphasized

  • production quality run-time inference infrastructure
  • building ML training pipeline
  • runtime inference integration
  • on-device model optimizations

Other signals

  • end-to-end ML development pipeline
  • shipping ML-based features and products
  • runtime inference integration
  • on-device model optimizations