Skip to content
CV Rocket
Get started
How it worksThe enginePricingBlogContactSign in

Open nowPosted 60 days agoWe saw it 16 min after it went up

ML Researcher - Image / Video Diffusion

Krea13 open roles

Where
San Francisco
Work mode
On site
Get the CV for this job

From $25 per CV, paid once. No subscription.

Your applicationOpen nowML Researcher - Image / Video DiffusionKrea · San Francisco

  1. YouYes, apply to this one.

  2. CV RocketCV written for this posting.

  3. 25 readersRecruiter, hiring manager, skeptic. Round after round.

  4. CV RocketApplied on Krea's own form.

The reply lands in your private mailbox

3×more interviews than doing it yourself with ChatGPT.

The clock on this job

Early applications get read.

12.9% of postings close within 7 days. Measured by our own scanner across the market.

Share of postings closed within
  1. 2.2%1 day
  2. 4.7%3 days
  3. 12.9%7 days
  4. 21.0%14 days
  5. 35.6%30 days
  6. 58.5%60 days
This job: posted 60 days ago

The posting

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, and with Krea 3 https://www.krea.ai/blog/introducing-krea-3 - our new image editing model.

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

From $25, paid onceGet the CV for this job

What happens when you press

One press. We do the rest.

  1. A CV for this posting

    Written against Krea's own wording, from every piece of relevant proof in your profile.

  2. 25 readers review it

    Recruiter, hiring manager, skeptic and more read every draft, round after round. You get the best round.

    The review screen in CV Rocket: how each CV was read, round by round.

  3. We apply on Krea's form

    Our application engine gets through the hardest forms there are. Where a question needs you, AI suggests the best answer. Don't want us applying from our IP addresses? Use our Chrome extension: we apply straight from your own browser.

    An application in CV Rocket: every answer filled in on the employer's form.

  4. Every reply, sorted

    Krea's answer lands in your private mailbox, and we classify it on arrival: interview, question, rejection.

    The CV Rocket inbox: each employer reply classified as an interview, an action or a rejection.

  5. Reply with AI

    AI helps you write the email, checks it and sends it. We show you whether the recruiter read it.

  6. The interview in your calendar

    Full integration with your calendar. The invitation goes straight in.

    An interview invitation in the CV Rocket inbox, added to the candidate's calendar.
Get the CV for this job

From $25 per CV, paid once. No subscription.

Why it works

3×

more interviews than doing it yourself with ChatGPT.

ChatGPT writes a CV and never learns what happened to it. We see every reply. For each CV we know:

  • How it was written, and how the review scored it
  • When we applied, and how long after the posting went up
  • Which posting, which company, which city
  • Who got the interview, and who heard nothing

That is how we know which CVs get called.

Get the CV for this job

From $25 per CV, paid once. No subscription.

The numbers game

More applications. More interviews.

Every application goes out with its own CV, written for that posting and paid once. Send enough of them and the law of large numbers finds you the job.

Get the CV for this job
By hand5–10
With CV Rocket100
applications a day

Nearby

Live postings like this one

Same employer first, then the same role elsewhere.

Before you press

Straight answers

Get the CV for this job

From $25 per CV, paid once. No subscription.

What if my background isn't good enough?

We make the most of the background you have. The CV uses every piece of relevant proof your profile holds, and one of the 25 readers reads your whole profile and flags what the CV left out.

Do you really apply for me?

Yes, on the employer's own form, the hardest ones included. Where a question needs you, you answer it right there and AI suggests the best answer. Don't want us applying from our IP addresses? Use our Chrome extension: we apply straight from your own browser.

Is it a subscription?

No. You pay once per CV, from $25. Every application goes out with its own CV, written for that posting.

One job. One CV.
Paid once.

Pick the posting you want. We write for it, apply for you and catch the reply.

Get the CV for this job

From $25 per CV, paid once. No subscription.