stable-diffusion-jax | Python Ecosystem Directory

Statistics for this project are still being loaded, please check back later.

Diffusers training with mmengine

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

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

Karras et al. (2022) diffusion models for PyTorch

High-Resolution Image Synthesis with Latent Diffusion Models

[RSS 2023] Diffusion Policy Visuomotor Policy Learning via Action Diffusion

JAX implementation of VQGAN

The official PyTorch implementation of the paper "Human Motion Diffusion Model"

Diffusion base mining

Outpainting with Stable Diffusion on an infinite canvas

An implementation of simple diffusion in PyTorch (and JAX)

web UI for GPU-accelerated ONNX pipelines like Stable Diffusion, even on Windows and AMD

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hug...

Multiple dispatch over abstract array types in JAX.