Staff Research Scientist, Image & Video Diffusion (Equity)

Black Forest Labs; United States; Hybrid; USD 140,000 - 370,000; Full time

Job summary

Black Forest Labs is seeking a researcher to train large-scale diffusion models for image and video generation, exploring which architectural choices matter and how to scale training across distributed systems.

You will conduct rigorous ablations, optimize speed-quality tradeoffs, and fine-tune models for tasks like upscalers and inpainting/outpainting, while communicating findings to steer our Flux research agenda across Freiburg and SF offices and remote teams.

Qualifications

• Hands-on experience training large-scale diffusion models for image and video data.
• Experience fine-tuning diffusion models for upscalers, inpainting, outpainting.
• Deep understanding of evaluation metrics for image and video generative models.
• Strong proficiency in PyTorch and transformer architectures.
• Experience debugging distributed training issues and presenting research findings.

Responsibilities

• Train large-scale diffusion transformer models for image and video data.
• Rigorously ablates design choices and communicates results to guide research.
• Evaluate speed-quality tradeoffs of neural architectures in production contexts.
• Fine-tune diffusion models for upscalers, inpainting/outpainting, and related tasks.

Job description

Black Forest Labs is seeking a researcher to train large-scale diffusion models for image and video generation, exploring which architectural choices matter and how to scale training across distributed systems.

You will conduct rigorous ablations, optimize speed-quality tradeoffs, and fine-tune models for tasks like upscalers and inpainting/outpainting, while communicating findings to steer our Flux research agenda across Freiburg and SF offices and remote teams.