Alternatives and similar repositories for smalldiffusion

Users that are interested in smalldiffusion are comparing it to the libraries listed below

• helblazer811 / Diffusion-Explorer
  Interactive visualizations of the geometric intuition behind diffusion models.
  ☆906Updated 5 months ago
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆966Updated last year
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆298Updated last year
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆624Updated last year
• SonyResearch / micro_diffusion
  Official repository for our work on micro-budget training of large-scale diffusion models.
  ☆1,535Updated 11 months ago
• tonyduan / diffusion
  From-scratch diffusion model implemented in PyTorch.
  ☆98Updated last year
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆655Updated last year
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆275Updated last year
• rbalestr-lab / lejepa
  ☆730Updated this week
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆429Updated last year
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆1,101Updated last month
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆330Updated 3 weeks ago
• facebookresearch / flow_matching
  A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…
  ☆3,837Updated 2 months ago
• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆215Updated last year
• lumalabs / imm
  Official implementation of Inductive Moment Matching
  ☆566Updated 5 months ago
• alexiglad / EBT
  PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning
  ☆563Updated last month
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆85Updated 11 months ago
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆294Updated 6 months ago
• gmongaras / Diffusion_models_from_scratch
  Creating a diffusion model from scratch in PyTorch to learn exactly how they work.
  ☆387Updated 8 months ago
• Quentin-Anthony / torch-profiling-tutorial
  ☆532Updated 4 months ago
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,237Updated 6 months ago
• AlmondGod / tinyworlds
  A minimal implementation of DeepMind's Genie world model
  ☆1,061Updated 3 weeks ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆548Updated 3 weeks ago
• Animadversio / DiffusionFromScratch
  Rebuild the Stable Diffusion Model in a single python script. Tutorial for Harvard ML from Scratch Series
  ☆222Updated 10 months ago
• eje24 / iap-diffusion-labs
  Labs for MIT 6.S184/6.S975, IAP 2025
  ☆268Updated last week
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆491Updated last year
• ludocomito / learning-diffusion
  A practical guide to diffusion models, implemented from scratch.
  ☆164Updated this week
• KellerJordan / Muon
  Muon is an optimizer for hidden layers in neural networks
  ☆2,075Updated 3 weeks ago
• apple / ml-tarflow
  ☆305Updated 11 months ago
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆306Updated last year