Machine Learning Research Engineer at Apple in Location not listed

Video is at the core of nearly all Apple products, and as an engineer on our team, you will research practical ML algorithms and build with us the next generation of video technology. We are looking for a highly ambitious individual, who will flourish working on technically meaningful problems at the frontier of machine learning — including computer vision, image and video generation, representation learning, and compression. Your work will redefine the video experience for billions of users.

In this role you will work together with colleagues to develop ML-based video approaches for current and future Apple products. This position requires a highly self-directed individual, who is comfortable working at the cutting edge of research. Strong and analytical skills will be critical towards solving challenging problems in uncharted technical territories.

Research various components of ML-based approaches to image/video processing problems: invent models, design appropriate datasets and pipelines to train them, prototype new architectures to improve performance, study existing literature, and so on
Collaborate closely with team members to optimize the efficiency of the models and deploy them on particular hardware architectures
Work with cross-functional teams to achieve quality and computational requirements towards shipping the technology
Communicate: demonstrate the technology, present to leadership, discuss progress with colleagues, and so on

BS and a minimum of 3 years relevant industry experience
3+ years of hands-on experience in ML research, which can include Ph.D. work
Proven track record of success in deep learning, with publications in top ML/CV venues
Familiarity with the latest ML and CV innovations, including GenAI modeling techniques
Proficency in Python and Software skills in common ML tools such as PyTorch or TensorFlow
Knowledge of ML and specifically deep learning: principles, model prototyping and architecture design, training procedures, visualization and debugging methodology, objective function design, and so on

Ph.D. or Master’s degree specializing in ML
Experience implementing custom ops in CUDA
Experience in designing practical ML implementations and having ML experience in industry
Experience with generative modeling, optical flow, perceptual quality, ML for video or data compression
Excellent written and oral communication skills