A

Apple

Designs phones, computers and their operating systems

Generative AI Engineer - Image/Video Restoration

Computer VisionConfirmed live in the last 24 hours
Full-timeBeijing, ChinaNo salary listedIn-person

About the Job

Requirements

  • A Master's or Ph.D. in Computer Science, Electrical Engineering, Applied Mathematics, or a related field.
  • Strong background in computer vision and deep learning, with proven experience in generative models such as diffusion models, generative adversarial networks, and transformers.
  • Proficiency in Python and deep learning frameworks; PyTorch is preferred.
  • Experience with large-scale image and video datasets and distributed training across multiple GPUs or nodes.
  • A solid understanding of image and video restoration metrics, including PSNR, SSIM, and LPIPS, and perceptual evaluation.
  • Hands-on experience with video preprocessing, including motion estimation and optical flow, frame alignment and stabilization, temporal consistency techniques, and video encoding and decoding.
  • Strong software engineering skills, including clean code, Git, debugging, and optimization.

Responsibilities

  • Research, design, and implement generative AI models for image and video restoration, including deblurring, denoising, super-resolution, inpainting, and frame interpolation.
  • Build and optimize training pipelines for large-scale datasets, including preprocessing, augmentation, and distributed training.
  • Evaluate restoration performance using objective metrics such as PSNR, SSIM, and LPIPS, as well as subjective and perceptual quality measures.
  • Develop scalable and efficient inference pipelines, optimizing latency, throughput, and memory.
  • Stay current with the latest research in computer vision and generative AI, and translate novel ideas into practical solutions.
  • Collaborate with cross-functional teams to integrate restoration models into production systems.

Desired Qualifications

  • A track record of publications or open-source contributions in generative AI, computer vision, or image and video restoration.
  • Experience with real-world video data processing, such as raw-domain data, HDR pipelines, and ISP sharpening.
  • Familiarity with cloud-based large-scale dataset management, such as S3 and distributed file systems.
  • Experience with real-time or near-real-time video restoration and performance optimization.
  • Knowledge of advanced motion analysis, including scene change detection, temporal consistency checks, and optical flow with RAFT or PWC-Net.
  • Familiarity with deployment on diverse hardware, including edge devices, mobile devices, and GPU acceleration.
  • Practical experience with efficient model deployment, including model compression, quantization, distillation, and hardware optimization such as TensorRT and mixed precision.

About the Company

Apple

Designs phones, computers and their operating systems

Apple designs and sells consumer electronics, from iPhone smartphones and Mac computers to tablets and wearable devices. Its operating systems manage apps and built-in functions for communication and work, while iCloud stores files and synchronizes data across devices. Customers buy additional digital services through subscriptions such as Apple Music and Apple TV, and developers distribute software through the App Store. Apple earns money from hardware sales, subscriptions and digital transactions, selling directly to customers and through retailers and mobile carriers around the world. It was founded in 1976 and has its headquarters in Cupertino, California.

Company size
10,001+
Company stage
IPO
Headquarters
Cupertino, CA
Founded
1976

Recently Posted Jobs

Explore more opportunities at Apple

See more jobs