Machine learning research engineer on Apple's Intelligent System Experiences team, building the generative model technology behind Genmoji, Image Playground, and Image Wand. Day to day: training and fine-tuning diffusion and auto-regressive image/video foundation models, from web-scale data and distributed training to on-device optimization, using frameworks like PyTorch, TensorFlow, or JAX.
Do you believe that generative models can fundamentally change how people interact with devices? Are you excited by the possibility of connecting people to unprecedented levels of creativity? Come join the team that turned cutting edge Generative AI research into compelling user experiences through Apple Intelligence. We are part of the Intelligent System Experiences team at Apple and work on the core modeling technology behind Genmoji, Image Playground and Image Wand. From web-scale datasets to distributed training of foundation models, application specific adapters to on-device optimizations, you will have the unique opportunity to advance and ship state-of-the-art technology that impacts Apple users around the world!
We are seeking a machine learning research engineer with experience building modern generative models based on diffusion and auto-regressive technologies. The ideal candidate has solid ML fundamentals, experience with applied research, and hands-on knowledge training and adapting large scale image/video/audio/multimodal foundation models. You will stay at the forefront of the latest AI research to rapidly apply impactful developments while simultaneously pioneering proprietary, experimental ideas designed specifically for Apple’s unique ecosystem.
Minimum Qualifications
M.S. or PhD in Computer Science or a related field such as Electrical Engineering, Robotics, Statistics, Applied Mathematics, or equivalent experience.
2+ years of experience working on training and fine-tuning modern image / video generation models
Experience working in an academic or industrial research environment
Proficient in an ML framework such as PyTorch, TensorFlow, JAX
Excellent communication skills and an ability to drive clarity in ambiguous circumstances
Preferred Qualifications
Track record of strong industrial research, including delivery of complex projects or publications in top-tier venues