Repository navigation

JaxDDPM

Public

About

A DDPM implementation in Jax for continuous space modeling.

10 stars

1 watching

Report repository

Name		Name	Last commit message	Last commit date
.gitignore		.gitignore
LICENSE		LICENSE
README.md		README.md
datasets.py		datasets.py
diffusion.py		diffusion.py
ensemble.py		ensemble.py
img_utils.py		img_utils.py
mlp.py		mlp.py
mlp_resnet_arch.py		mlp_resnet_arch.py
planes.py		planes.py
power.py		power.py
train.py		train.py

Repository files navigation

JaxDDPM

A DDPM implementation in Jax for continuous space modeling. Complementary to https://github.com/philippe-eecs/IDQL.

Credit to Ilya Kostrikov for help in making this repo!

About

A DDPM implementation in Jax for continuous space modeling.

10 stars

1 watching

Report repository

Releases

Packages

Contributors

Languages