Member of Technical Staff - Image / Video Researcher Remote | Ger...
Tbwa Chiat/Day Inc - MS, United States
Work at Tbwa Chiat/Day Inc
Overview
- View job
Overview
Member of Technical Staff - Image / Video Researcher
Remote | Germany | USA
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.
Role:
- 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
Ideal Experiences:
- Training large scale Diffusion models for image and video data
- Finetuning Diffusion models for image and video applications, such as image and video upscalers, in and out painting models, etc.
- 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
Nice to have:
- Experience with writing forward and backward Triton kernels and ensuring their correctness while considering floating point errors
- Profiling, debugging, and optimizing single and multi-GPU operations using tools such as Nsight or stack trace viewers