simple-diffusion

Versatile-Diffusion

Versatile Diffusion: Text, Images and Variations All in One Diffusion Model, arXiv 2022 / ICCV 2023

audio-diffusion-pytorch

Audio generation using diffusion models, in PyTorch.

stable-diffusion-jax

med-seg-diff-pytorch

Implementation of MedSegDiff in Pytorch - SOTA medical segmentation using DDPM and filtering of f...

v-diffusion-jax

v objective diffusion inference code for JAX.

vqgan-jax

JAX implementation of VQGAN

bit-diffusion

Implementation of Bit Diffusion, Hinton's group's attempt at discrete denoising diffusion, in Pyt...

diff-mining

Diffusion base mining

DiT

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

diffrax

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable. https://docs....

manifold-diffusion

Diffusion on manifolds for image retrieval

k-diffusion

Karras et al. (2022) diffusion models for PyTorch

vit-vqgan

JAX implementation ViT-VQGAN

vrp-attention-jax

Reimplementation of the paper "Attention, Learn to Solve Routing Problems!" in jax/flax.

stable-dreamfusion

Text-to-3D & Image-to-3D & Mesh Exportation with NeRF + Diffusion.