---
title: 'ML Researcher - Image / Video Diffusion at Krea'
canonical: 'https://feeny.ai/job/ml-researcher-image-video-diffusion-krea-san-francisco-q0z8r0s4arnw'
type: 'job'
last_seen: '2026-09-08'
---

# ML Researcher - Image / Video Diffusion at Krea

- **Company:** Krea
- **Location:** San Francisco, CA
- **Employment:** full-time
- **Work type:** onsite
- **Posted:** 2026-09-01
- **Last confirmed live:** 2026-09-08
- **Apply:** https://jobs.ashbyhq.com/krea/845dce21-e270-49a2-9229-5becd256b829

## Job description

## ABOUT KREA

At Krea, we are building next-generation AI creative tools.

We're dedicated to making AI intuitive and controllable for creatives - our mission is to build tools that empower human creativity, not replace it. We believe AI is a new medium that allows us to express ourselves through various formats - text, images, video, sound, and even 3D. We're building better, smarter, and more controllable tools to harness this medium. We recently took this a step forward with the launch of Krea 2 https://www.krea.ai/blog/krea-2-image-model, our first foundation model, built completely from scratch for aesthetic diversity and stylistic control.

We've raised over $83M and are backed by world-class investors such as a16z, Bain Capital, and Abstract. We work full-time and in-person at our waterfront office in San Francisco. We care about creativity: our team includes musicians, designers, visual artists, and engineers.

We're looking for an experienced Researcher with engineering skills who can work on large-scale image and video models training experiments, with experience training image models at scale.

## OUR CULTURE

- We work full-time and in-person at our North Beach office in San Francisco.
- We believe that demonstrated interest in the creative space is key: our team includes musicians, designers, visual artists and more.
- Fast iteration and execution speed. Bias towards action, agency, and independence.

## WHAT YOU'LL DO

- Train diffusion models for image and video generation on large GPU clusters.
- Fully optimize and profile large distributed training runs across model architectures, kernels, data loading, memory constraints, and communication.
- Implement and improve various distributed training strategies including FSDP, CP, SP, TP, and EP.
- Continuously improve model quality and reliability through data, model architecture, training pipeline, structuring experiments, and eval design.
- Debug distributed training errors and implement fault tolerance solutions, identifying bad GPU, NVLink, Infiniband (IB) components as well as monitoring numerical errors and NCCL issues.
- Ablate different architecture, attention, optimizer, data, and algorithmic choices to reliably improve efficiency and performance of our models.

## WHAT WE'RE LOOKING FOR

- Proven track record in working with image or video models at scale (publications or open-source contributions a plus).
- Strong proficiency in PyTorch and understanding of its inner workings.
- Strong background in distributed training paradigms such as FSDP, CP, SP, USP, TP, and EP. Knowing how different parallelism strategies work together and their tradeoffs.
- Experience in profiling and debugging large distributed training. Being comfortable with analyzing traces to identify bottlenecks and look for improvements.
- Good knowledge of low precision training / inference in FP8, NVFP4, and MXFP8.
- Solid understanding of diffusion model training pipeline across pretraining, midtraining, preference optimization, and reinforcement learning.
- Keeping up with the developments in related fields such as LLM, VLM, representation learning, and robotics research.
- Being comfortable working in a goal-oriented research environment.
- Having good judgement around when one should explore different training strategies and when it's time to commit to a specific strategy to scale compute and data.
- Comfortable working with underspecified goals. We expect every technical member to take an ambiguous research goal and break it down into concrete requirements, plans, experiment plan, and execution items.
- Good research taste — bias towards simplicity and methods that scale well with compute, data, and minimal human supervision.
- Ability to iterate rapidly, and propose creative research directions.
- Be comfortable getting your hands dirty with data and designing custom data pipelines to improve data quality.

## WHAT WE OFFER

- Team: Work alongside a world-class team building the future of AI creative tooling
- Impact: Significant scope and company-wide impact
- Competitive compensation: generous salary & equity packages
- Health & wellness: 100% health & 99% dental/vision insurance premiums covered for employees, health FSA accounts, & long-term disability coverage
- Time off: Flexible PTO policy
- Financial planning: 401k with a 4% company-sponsored match
- Meals in the office: breakfast, lunch, dinner - you name it, we'll cover it
- Transit: Ubers covered to & from the office
- Sponsorship: We're open to sponsoring international visas where we can (e.g., STEM OPT, OPT, H-1B, O-1, E-3).
- And more!

Please note the above benefits & perks are for full-time employees

## About Krea

## Company Overview
- **One-liner**: Krea builds an AI-powered creative suite for generating, editing, and enhancing images, video, and 3D assets in real time.
- **Entity Type**: Private (Series B – $83M total funding)
- **Headquarters**: San Francisco, California, United States
- **Founded**: Not publicly available (blogs indicate operations began roughly 2023–2024)
- **Founders**: Victor Perez, Diego Rodriguez

## Core Business
- **Industries**: Creative AI, generative media, artificial intelligence
- **Target customers**: B2C (individual creatives, enthusiasts) and B2B (enterprise teams at companies like Pixar, LEGO, Samsung, Shopify, Microsoft)
- **Mission / purpose**: “Make AI feel intuitive and natural by giving creators full control over their process” and “multiply human creativity”

## Products & Services
- **Krea AI Suite (Image Generation)**: Native 4K image generation with 1,000+ styles, 20+ models, 3-second generation speed for a 1024px Flux image.
- **AI Video Generation**: Access to top models (Veo 3, Kling, Hailuo, Wan, Runway) with intuitive interface for social media clips, animations, and edits.
- **AI 3D Generation**: Text-to-3D and 3D object manipulation within images.
- **Real-Time Rendering**: Market‑leading real‑time image generation (<50ms) and real‑time video with frame consistency.
- **Image & Video Upscaling**: Upscale images up to 22K resolution and videos up to 8K at 120fps (includes Topaz Photo/Gigapixel).
- **LoRA Fine-tuning**: Train custom image or video models with a few images for style, character, or object consistency.
- **Generative Editing**: Add/remove objects, merge images, change expressions or lighting using 10+ editing models.
- **Enterprise Product (upcoming)**: SOC 2 Type II and ISO 27001 compliant, designed for enterprise teams.

All offerings are SaaS, accessible via web browser.

## Market Standing
- **Valuation / Market Cap**: Not disclosed
- **Key Metric**: **$83M total funding** (Series B led by Bain Capital Ventures, April 2025)
- **Notable Investors / Partners**: Bain Capital Ventures (Series B lead), Andreessen Horowitz (Series A), Pebblebed (seed), Abstract Ventures (seed), plus early backers HF0, A*, Gradient Ventures. Enterprise clients include Pixar, LEGO, Samsung, Shopify, Microsoft.
- **Growth Signals**:
  - 30+ million users across 191 countries (organic growth)
  - 20x revenue growth on a team of 8 originally
  - 50+ major product updates rolled out
  - Team expanded from 8 to 17 (as of April 2025); new ocean-view office in SF
  - Adoption by major studios and brands

## Competitive Advantages
- **Model aggregation**: One subscription unlocks 64+ models (Flux, Wan2.1, Veo 2, Kling, Hailuo, etc.) – no need to maintain separate subscriptions.
- **Industry‑fastest inference**: 3s for a 1024px Flux image at FP16; real‑time generation <50ms.
- **Interconnected workflows**: Generate an image → upscale → convert to video → add sound – all in one tool, with editable “knobs and dials” at every step.
- **Community‑driven development**: Ships new features every Thursday; product built by listening to users first, then building.
- **Integrated LoRA training**: Train custom models directly in the browser without hosting weights externally.

## Strategic Focus
- **Research**: Krea AI Research Lab focusing on aesthetics, controllability, and hyper‑personalization.
- **Product roadmap**: Release of a new generation of aesthetic models with deeper artistic understanding and novel control modules that go beyond text prompts.
- **Enterprise expansion**: Launch enterprise product with compliance certifications (SOC 2, ISO 27001).
- **Multimodal future**: Developing infrastructure for generating immersive, real‑time environments (like videogames).

## Why Work Here
- **Culture**: Small, highly talented team (currently 17) that lives and breathes creativity – founders and many team members are musicians, poets, designers, videographers; “The team behind Krea is the work of a small, talented group of imaginative, incredibly dedicated people.”
- **Engineering & research**: Engineers from MIT, Princeton, UC Berkeley, University of Chicago; authors of state‑of‑the‑art models (Lumina, Anifusion). Freedom to do own research after building products for the community.
- **Work environment**: New office with Pacific Ocean views in San Francisco; majority of team shares a living space (noted early‑stage).
- **Velocity**: Weekly Thursday releases, exposure to cutting‑edge AI models, direct impact on a product used by millions.
- **Open roles**: Listed at [jobs.ashbyhq.com/krea](https://jobs.ashbyhq.com/krea). Ideal candidate is “deeply technical, artistically opinionated.”

## Sources
1. [krea.ai – Product & Pricing](https://www.krea.ai/)
2. [Bloomberg – Krea AI Inc Company Profile](https://www.bloomberg.com/profile/company/2562987D:US)
3. [Krea Blog – “A New Krea” (Series B announcement)](https://www.krea.ai/index/new-krea)
4. [Andreessen Horowitz – “Investing in Krea”](https://a16z.com/announcement/investing-in-krea/)
5. [Krea Careers Page](https://jobs.ashbyhq.com/krea)

## Other roles at Krea

- [ML Researcher - Posttraining](https://feeny.ai/job/ml-researcher-posttraining-krea-san-francisco-je648m0mv2k4) — San Francisco, CA
- [Product Engineer](https://feeny.ai/job/product-engineer-krea-san-francisco-a1c64g6r1rzn) — San Francisco, CA
- [Fullstack Engineer](https://feeny.ai/job/fullstack-engineer-krea-san-francisco-s725r01bry0f) — San Francisco, CA
- [Founding Customer Success Manager](https://feeny.ai/job/founding-customer-success-manager-krea-san-francisco-n3a6rmtt84x2) — San Francisco, CA
- [Account Executive, API + Open Source](https://feeny.ai/job/account-executive-api-open-source-krea-san-francisco-859tkyrr1gz4) — San Francisco, CA
- [Software Engineer, Backend](https://feeny.ai/job/software-engineer-backend-krea-san-francisco-m7qrcgfqcrp2) — San Francisco, CA
- [Creative Lead](https://feeny.ai/job/creative-lead-krea-san-francisco-c2v2y0dcfzwv) — San Francisco, CA
- [Engineer, Supercomputing & Distributed Systems](https://feeny.ai/job/engineer-supercomputing-distributed-systems-krea-san-francisco-pb5gr7p7bw1f) — San Francisco, CA
- [Account Executive](https://feeny.ai/job/account-executive-krea-san-francisco-padwn23g3rbp) — San Francisco, CA
- [ML Engineer - Personalization & Recommendation Systems](https://feeny.ai/job/ml-engineer-personalization-recommendation-systems-krea-san-francisco-k0pnwba3rmtc) — San Francisco, CA