Researcher in Generative Image and Video Models - Remote

Overview

Black Forest Labs is a cutting-edge startup pioneering generative image and video models. Our team, which invented Stable Diffusion, Stable Video Diffusion, and FLUX.1, is currently seeking a strong researcher to work on model training and development.

In Short

• 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

Requirements

• Training large scale Diffusion models for image and video data
• Finetuning Diffusion models for image and video applications
• Deep understanding of how to effectively evaluate image and video generative models
• Strong proficiency in PyTorch, transformer models and other NN architectures
• Deep understanding of training techniques such as FSDP, low precision training, and model parallelism

Benefits

• Experience with writing forward and backward Triton kernels
• Profiling, debugging, and optimizing single and multi-GPU operations