simple-diffusion

An implementation of simple diffusion in PyTorch (and JAX)

MIT License

Stars
18
View Code on GitHub View on X
Ecosystems: Python

[WiP]simple-diffusion

An implementation of simple diffusion: End-to-end diffusion for high resolution images in PyTorch (and maybe JAX)

Related Projects
stable-diffusion-jax

12 Aug 2022 85
DiT

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

16 Dec 2022 6,050
audio-diffusion-pytorch

Audio generation using diffusion models, in PyTorch.

07 Jul 2022 1,929
k-diffusion

Karras et al. (2022) diffusion models for PyTorch

21 Jun 2022 2,277
diff-mining

Diffusion base mining

27 Apr 2024 37
manifold-diffusion

Diffusion on manifolds for image retrieval

24 May 2018 124
stable-dreamfusion

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

06 Oct 2022 8,194
diffrax

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

25 Jul 2021 1,390
bit-diffusion

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

17 Aug 2022 330
Versatile-Diffusion

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

02 Nov 2022 1,312
vit-vqgan

JAX implementation ViT-VQGAN

22 Jun 2022 77
v-diffusion-jax

v objective diffusion inference code for JAX.

07 Nov 2021 209
vqgan-jax

JAX implementation of VQGAN

28 Jun 2021 90
med-seg-diff-pytorch

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

23 Nov 2022 209
vrp-attention-jax

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

03 Oct 2021 7