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.
  ☆897Updated 5 months ago
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆958Updated last year
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆296Updated last year
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆617Updated last year
• tonyduan / diffusion
  From-scratch diffusion model implemented in PyTorch.
  ☆98Updated last year
• SonyResearch / micro_diffusion
  Official repository for our work on micro-budget training of large-scale diffusion models.
  ☆1,523Updated 10 months ago
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆326Updated last week
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆84Updated 10 months ago
• rbalestr-lab / lejepa
  ☆562Updated last week
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆272Updated last year
• lumalabs / imm
  Official implementation of Inductive Moment Matching
  ☆561Updated 4 months ago
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆653Updated last year
• eje24 / iap-diffusion-labs
  Labs for MIT 6.S184/6.S975, IAP 2025
  ☆259Updated this week
• Animadversio / DiffusionFromScratch
  Rebuild the Stable Diffusion Model in a single python script. Tutorial for Harvard ML from Scratch Series
  ☆217Updated 10 months ago
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆300Updated last year
• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆214Updated last year
• facebookresearch / flow_matching
  A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…
  ☆3,737Updated last month
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆790Updated 11 months ago
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆427Updated 11 months ago
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆1,077Updated 2 weeks ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆547Updated this week
• Quentin-Anthony / torch-profiling-tutorial
  ☆528Updated 3 months ago
• gmongaras / Diffusion_models_from_scratch
  Creating a diffusion model from scratch in PyTorch to learn exactly how they work.
  ☆385Updated 7 months ago
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,229Updated 6 months ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆489Updated last year
• google-deepmind / nanodo
  ☆285Updated last year
• facebookresearch / jepa-intuitive-physics
  This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"
  ☆196Updated 9 months ago
• LambdaLabsML / distributed-training-guide
  Best practices & guides on how to write distributed pytorch training code
  ☆540Updated last month
• microsoft / dion
  Dion optimizer algorithm
  ☆384Updated this week
• alexiglad / EBT
  PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning
  ☆558Updated last week