Member of Technical Staff - Image / Video Generation

Job summary

Responsibilities

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.

• 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…
• 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

Tech stack

• PyTorch
• Triton