Member of Technical Staff - Image / Video Researcher

View full description

Requirements:

• Training large scale Diffusion models for image and video data.
• Finetuning Diffusion models for applications like upscalers and in/out painting models.
• Deep understanding of evaluating image and video generative models.
• Strong proficiency in PyTorch and other NN architectures.
• Understanding of training techniques such as FSDP, low precision training, and model parallelism.

Responsibilities:

• Training large scale Diffusion (transformer) models for image and video.
• Rigorously ablating design choices and communicating results & decisions with the broader team.
• Reasoning about the speed and quality trade-offs of neural network architectures.