# Member of Technical Staff - Image / Video Generation

**Company:** [Black Forest Labs](https://hotfix.jobs/companies/black-forest-labs)
**Location:** Freiburg, Germany
**Role:** ML Engineering
**Salary:** €130k – €340k/yr
**Skills:** Diffusion Models, PyTorch, Transformers, Deep Learning, Fsdp, Low-Precision Training, Model Parallelism, Triton, Nsight, Distributed Training, Image Generation, Video Generation
**Posted:** 2026-08-27

> Trains and fine-tunes large-scale diffusion transformer models for image and video generation, conducts rigorous ablation studies, and optimizes distributed training. Requires hands-on diffusion-model experience, strong PyTorch and transformer expertise, and understanding of generative-model evaluation.

## Job Description

## Responsibilities
- Train large-scale diffusion transformer models for image and video data.
- Rigorously ablate design choices by isolating variables, controlling for confounds, and communicating findings to shape research direction.
- Evaluate speed-quality tradeoffs of neural network architectures in production settings.
- Fine-tune diffusion models for specialized applications, including image and video upscalers, inpainting, and outpainting.
- Debug distributed training issues and present research findings to the team.

## Requirements
- Hands-on experience training large-scale diffusion models for image and video data.
- Practical knowledge of common failure modes and important training considerations.
- Experience fine-tuning diffusion models for specialized applications.
- Deep understanding of evaluating image and video generative models and selecting meaningful quality metrics.
- Strong proficiency in PyTorch, transformer architectures, and modern deep learning.
- Solid understanding of distributed training techniques, including FSDP, low-precision training, and model parallelism.

## Nice-to-Haves
- Experience writing forward and backward Triton kernels, ensuring correctness while considering floating-point errors.
- Proficiency profiling, debugging, and optimizing single- and multi-GPU operations using tools such as Nsight and stack trace viewers.
- Knowledge of the performance characteristics of architectural choices at scale.
- Published research contributing to understanding of generative models.

## Compensation
- Base annual salary: €130,000–€340,000, plus equity.

## Similar jobs

- [AI Engineer - New Verticals](https://hotfix.jobs/jobs/4159c72e-536e-4211-969c-6bcb9ad805fd) - Protege - Remote
- [Agent Systems Engineer](https://hotfix.jobs/jobs/2b04a068-018b-4fe0-915f-4fdf80d4c582) - Adaption Labs - San Francisco, CA
- [Inference Performance Engineer](https://hotfix.jobs/jobs/7718c69a-6d49-445e-8c7c-840bd1b3f346) - Adaption Labs - San Francisco, CA
- [Member of Technical Staff - Post Training](https://hotfix.jobs/jobs/e6d896a4-75d3-41e0-ac75-1a73065b0f27) - Black Forest Labs - Freiburg, Germany - €130k – €340k/yr
- [Senior AI Engineer – Notebooks](https://hotfix.jobs/jobs/fed3b705-a67d-4a63-8cff-b6c19357b4f7) - Datadog - Bordeaux, France

**Apply:** https://hotfix.jobs/jobs/405a95a1-d8ff-49a2-9851-5a5e1cf05ad8
**Canonical:** https://hotfix.jobs/jobs/405a95a1-d8ff-49a2-9851-5a5e1cf05ad8