Official Code for Stable Cascade

[NeurIPS 2024 Oral][GPT beats diffusion🔥] [scaling laws in visual generation📈] Official impl. of "Visual Autoregressive Modeling: Scalable Image Generation via Next-Scale Prediction". An *ultra-sim…

Flax is a neural network library for JAX that is designed for flexibility.

A scikit-learn compatible neural network library that wraps PyTorch

Kandinsky 2 — multilingual text2image latent diffusion model

functorch is JAX-like composable function transforms for PyTorch.

Simple image captioning model

Official SRFlow training code: Super-Resolution using Normalizing Flow in PyTorch

PeRFlow: Piecewise Rectified Flow as Universal Plug-and-Play Accelerator (NeurIPS 2024)

Estimates the size of a PyTorch model in memory

A mini-library for training consistency models.

Its a implementation of DeepFont : Identify Your Font from An Image using Keras

Using pre-trained Diffusion models as priors for inference tasks

Posterior Sampling using Latent Diffusion