Senior Computer Vision and Machine Learning Engineer, Creativity Apps

Apple Apple · Big Tech · Cupertino, CA +1 · Software and Services

Senior Computer Vision and Machine Learning Engineer at Apple focused on Generative AI for creative editing tools. The role involves incubating, training, evaluating, and deploying ML models, particularly diffusion models, transformers, and GANs, with a focus on computational photography and multi-modal image editing. The engineer will also work on efficient inference and collaborate with cross-functional teams to bring these technologies to Apple products.

What you'd actually do

  1. Incubate state-of-the-art machine learning algorithms and demonstrate their applicability to Apple's products
  2. Take ownership of the full lifecycle of machine learning models, from large distributed training and validation to efficient inference at scale
  3. Design data pipelines in partnership with collection, curation and evaluation teams to ensure suitable model inputs
  4. Communicate the latest research developments in image generation to help drive the long-term product roadmap
  5. Partner with cross-functional teams to bring experimental models and proof-of-concept features to next-generation products

Skills

Required

  • MS or PhD in Computer Science, Machine Learning or related field
  • 5+ years of significant industry experience
  • deep knowledge in computational photography and multi-modal image editing
  • deep knowledge in theory and practice of machine learning and deep learning techniques
  • diffusion models
  • variational auto-encoders
  • transformers
  • Experience delivering customer-facing products with computer vision, computational photography or generative AI features
  • Hands-on experience building, training, evaluating, and deploying diffusion, transformer and Generative Adversarial Network based models, or related methods
  • Experience contributing to large codebases
  • delivering high-quality software at scale
  • Strong programming skills in Python
  • PyTorch, JAX, or TensorFlow
  • Ability to collaborate effectively across organizations
  • drive alignment in large cross-functional projects
  • Ability to concisely communicate with audiences of different backgrounds

Nice to have

  • optimizing models and algorithms that run efficiently on resource-constrained platforms
  • flow matching
  • multi-modal conditioning for generative modeling
  • 3D reconstruction techniques from single or multiple images, particularly ML-based methods
  • Publications at major conferences (CVPR, ICCV, ECCV, NeurIPS, ICML, ICLR)

What the JD emphasized

  • delivering products using state-of-the-art computational photography and machine learning technologies
  • delivering customer-facing products with computer vision, computational photography or generative AI features
  • building, training, evaluating, and deploying diffusion, transformer and Generative Adversarial Network based models, or related methods
  • delivering high-quality software at scale

Other signals

  • Generative AI
  • diffusion models
  • transformers
  • GANs
  • computational photography
  • multi-modal image editing