Quick overview
Job description
About Black Forest Labs
We’re the team building the foundational layer for visual intelligence: multimodal models trained jointly on images, video, audio, and action to see, generate, and understand the world.
Our founding team pioneered latent diffusion and Stable Diffusion, the breakthroughs that made generative AI accessible to millions. With FLUX, we're pushing into generative media at scale, machines that act in the physical world, and emerging applications in computer use.
Our FLUX models top public benchmarks and run in production in some of the world's biggest platforms. We're a small, fast growing research lab competing with the largest companies in the world. From our labs in Freiburg to San Francisco, we're building what comes next.
Why This Role
You'll train large-scale diffusion models for image and video generation, exploring new approaches while maintaining the rigor that helps us distinguish meaningful progress from incremental tweaks. This isn't about following established recipes—it's about running the experiments that clarify which architectural choices matter and which are less impactful.
What You’ll Work On
Trains large-scale diffusion transformer models for image and video data, working at the scale where intuitions break and empirical evidence matters
Rigorously ablates design choices—running experiments that isolate variables, control for confounds, and produce insights you can actually trust—then communicating those results to shape our research direction
Reasons about the speed-quality tradeoffs of neural network architectures in production settings where both constraints matter simultaneously
Fine-tunes diffusion models for specialized applications like image and video upscalers, inpainting/outpainting models, and other tasks where general-purpose models aren't enough
What We’re Looking For
You've trained large-scale diffusion models and developed strong intuitions about what matters. You know that at research scale, every design choice has tradeoffs, and the only way to know which ones are worth making is through careful ablation. You're comfortable debugging distributed training issues and presenting research findings to the team.
You likely have:
Hands-on experience training large-scale diffusion models for image and video data, with practical knowledge of common failure modes and what matters most in training
Experience fine-tuning diffusion models for specialized applications—upscalers, inpainting, outpainting, or other tasks where understanding the domain matters as much as understanding the architecture
Deep understanding of how to effectively evaluate image and video generative models—knowing which metrics correlate with quality and which are just convenient proxies
Strong PyTorch fluency; you write research code that others can build on. You work fluently with coding agents to move faster, and have the depth to verify their output and own the result.
Experience running multi-agent workflows: you know which tasks to hand off and which to keep yourself, and you use the speed-up to raise the quality bar, not lower it.
Solid understanding of distributed training techniques—FSDP, low precision training, model parallelism—because our models don't fit on one GPU and training decisions impact research outcomes
We'd be especially excited if you:
Have experience writing forward and backward Triton kernels and ensuring their correctness while considering floating point errors
Bring proficiency with profiling, debugging, and optimizing single and multi-GPU operations using tools like Nsight or stack trace viewers
Know the performance characteristics of different architectural choices at scale
Have published research that contributed to how people think about generative models
How We Work Together
We’re a distributed team with real offices that people actually use. This role is based in Freiburg, and you’ll join us in the office at least 2 days a week, or one full week every other week.
Everything we do is grounded in four values:
Obsessed. We are a frontier research lab. The science has to be right, the understanding deep, the product beautiful.
Low Ego. The work speaks. The best idea wins, no matter who said it. Credit is shared. Nobody is above any task.
Bold. We take the ambitious bet. We ship, we do not wait for conditions to be perfect.
Kind. People over politics. We treat each other with genuine warmth. Agency without empathy creates chaos.
If this sounds like work you’d enjoy, we’d love to hear from you.
Base Annual Salary:
EU €130,000-€340,000 + Equity
Note: We may use AI-assisted tools to help organize applications. All hiring decisions are always made by our team.
Haystack's AI agent can fill in and submit this application for you. How AI auto apply works
Similar jobs
- CL

Founding Engineer, NLP Systems and InfrastructureRecently posted Auto Apply
Clera · Munich · On-site · Posted 12 Nov 2026 🇩🇪
DockerNLPHelm+6Salary not listed - CL

Robot Learning EngineerRecently posted Auto Apply
Clera · Munich · On-site · Posted 9 Nov 2026 🇩🇪
RoboticsComputer VisionDeep Learning+5EngineeringSalary not listed - BJ
Member of Technical Staff, Machine LearningRecently posted Auto Apply
Bjakcareer · Germany · Remote · Posted 10 Nov 2026 🇩🇪
LLMPyTorchPython+1TechnologySalary not listed - ST
Machine Learning Engineer (All genders)Recently posted Auto Apply
STARK · Munich · Hybrid · Posted 11 Nov 2026 🇩🇪
GCPSQLAWS+10TechnologySalary not listed - SI
Staff MLOps Engineer (m/f/d) Auto Apply
SIXT · Munich, BY · Hybrid · Posted 6 Nov 2026 🇩🇪
Generative AIEngineeringSalary not listed - ZE

MLOps Engineer, Deep Learning Auto Apply
Zenseact · München · Hybrid 🇩🇪
MLOpsDeep LearningDiversity & Inclusion+3EngineeringSalary not listed - VO

Senior ML Engineer - NLP (m/f/d) Auto Apply
Voize · Berlin · Hybrid · Posted 6 Nov 2026 🇩🇪
NLPKotlinLLM+2TechnologySalary not listed - TG
Senior AI/ML Engineer Auto Apply
team.blue Global · Berlin · Remote · Posted 20 Oct 2026 🇩🇪
DockerMLOpsMLflow+8TechnologySalary not listed - DO

Principal Machine Learning Engineer - Applied AI & LLMs (x/f/m) Auto Apply
Doctolib · Berlin · Hybrid · Posted 2 Nov 2026 🇩🇪
GCPSwiftAWS+17TechnologySalary not listed - SI
Working Student (f/m/d) – Development for AI Use Cases Auto Apply
SAP IT Business Systeme · Walldorf · Hybrid · Posted 4 Nov 2026 🇩🇪
Generative AITechnologySalary not listed - NA

Lead Machine Learning Engineer - Geometric Spatial AI (F/M/D) Auto Apply
NavVis · Munich Hybrid (NavVis GmbH) · Hybrid · Posted 30 Oct 2026 🇩🇪
LinearMLflowScikit-learn+10TechnologySalary not listed - PG
Senior Machine Learning Engineer (m/w/d) Auto Apply
Proalpha Group · Germany · Hybrid · Posted 30 Oct 2026 🇩🇪
NLPComputer VisionDeep Learning+3TechnologySalary not listed
