Back
News
August 8th, 2022
minute read

Running JAX Diffusion Models on Runpod

Zhen Lu
Zhen Lu

Heading 1

Heading 2

Heading 3

Heading 4

Heading 5
Heading 6

Lorem ipsum dolor sit amet, consectetur adipiscing elit, sed do eiusmod tempor incididunt ut labore et dolore magna aliqua. Ut enim ad minim veniam, quis nostrud exercitation ullamco laboris nisi ut aliquip ex ea commodo consequat. Duis aute irure dolor in reprehenderit in voluptate velit esse cillum dolore eu fugiat nulla pariatur.

Block quote

Ordered list

  1. Item 1
  2. Item 2
  3. Item 3

Unordered list

  • Item A
  • Item B
  • Item C

Text link

Bold text

Emphasis

Superscript

Subscript

Here's a quick and dirty guide to getting Huemin's JAX diffusion to run on runpod!

Today I figured I'd switch things up a bit and leave you a gif quick start instead of the usual article. It only takes a few minutes to get started with this and all the deps have been sorted out for you in the nb. If you need the notebook file that you should upload to your instance, you can download it from one of these two places: Link 1, Link 2

Please note that it will sit here for quite a few minutes while it downloads models and prepared for the first render cycle:

Be patient and it will start rendering when it's ready. I was getting around 1.5-2s/it on the demo example.

Enjoy!

Credits to: Huemin (@huemin_art), for the notebook, Alexander Redde (@alexanderredde3) for working out the deps, and nsheppherd (@nshepperd1) and Rivers Have Wings (@rivershavewings) for Jax Diffusion

Newly  Features

We've cooked up a bunch of improvements designed to reduce friction and make the.

Create ->
Newly  Features

We've cooked up a bunch of improvements designed to reduce friction and make the.

Create ->
Newly  Features

We've cooked up a bunch of improvements designed to reduce friction and make the.

Create ->
Newly  Features

We've cooked up a bunch of improvements designed to reduce friction and make the.

Create ->
We're officially SOC 2 Type II Compliant
You've unlocked a referral bonus! Sign up today and you'll get a random credit bonus between $5 and $500
You've unlocked a referral bonus!
Claim Your Bonus
Claim Bonus
Blog

Running JAX Diffusion Models on Runpod

Curious about JAX-based diffusion models? This post walks through setting up and running them on Runpod using our GPU pods. It covers environment setup, model launching, and highlights the performance benefits of JAX for image generation workflows.

Author
Zhen Lu
Date
August 8, 2022
Table of contents
Share
Running JAX Diffusion Models on Runpod

Here's a quick and dirty guide to getting Huemin's JAX diffusion to run on runpod!

Today I figured I'd switch things up a bit and leave you a gif quick start instead of the usual article. It only takes a few minutes to get started with this and all the deps have been sorted out for you in the nb. If you need the notebook file that you should upload to your instance, you can download it from one of these two places: Link 1, Link 2

Please note that it will sit here for quite a few minutes while it downloads models and prepared for the first render cycle:

Be patient and it will start rendering when it's ready. I was getting around 1.5-2s/it on the demo example.

Enjoy!

Credits to: Huemin (@huemin_art), for the notebook, Alexander Redde (@alexanderredde3) for working out the deps, and nsheppherd (@nshepperd1) and Rivers Have Wings (@rivershavewings) for Jax Diffusion

Build what’s next.

The most cost-effective platform for building, training, and scaling machine learning models—ready when you are.

You’ve unlocked a
referral bonus!

Sign up today and you’ll get a random credit bonus between $5 and $500 when you spend your first $10 on Runpod.