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.
  ☆822Updated 3 months ago
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆934Updated last year
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆291Updated last year
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆610Updated last year
• tonyduan / diffusion
  From-scratch diffusion model implemented in PyTorch.
  ☆97Updated last year
• SonyResearch / micro_diffusion
  Official repository for our work on micro-budget training of large-scale diffusion models.
  ☆1,507Updated 8 months ago
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆648Updated last year
• Quentin-Anthony / torch-profiling-tutorial
  ☆497Updated last month
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆257Updated last year
• eje24 / iap-diffusion-labs
  Labs for MIT 6.S184/6.S975, IAP 2025
  ☆216Updated 4 months ago
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆293Updated 2 months ago
• gmongaras / Diffusion_models_from_scratch
  Creating a diffusion model from scratch in PyTorch to learn exactly how they work.
  ☆377Updated 5 months ago
• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆208Updated last year
• lumalabs / imm
  Official implementation of Inductive Moment Matching
  ☆544Updated 2 months ago
• Animadversio / DiffusionFromScratch
  Rebuild the Stable Diffusion Model in a single python script. Tutorial for Harvard ML from Scratch Series
  ☆216Updated 7 months ago
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,209Updated 3 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆72Updated 8 months ago
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆427Updated 9 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆536Updated last week
• microsoft / dion
  Dion optimizer algorithm
  ☆343Updated 2 weeks ago
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆292Updated 10 months ago
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆1,018Updated 5 months ago
• alexiglad / EBT
  PyTorch Code for Energy-Based Transformers paper -- generalizable reasoning and scalable learning
  ☆502Updated 2 weeks ago
• facebookresearch / flow_matching
  A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…
  ☆3,394Updated last week
• facebookresearch / jepa-intuitive-physics
  This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"
  ☆183Updated 7 months ago
• kvablack / ddpo-pytorch
  DDPO for finetuning diffusion models, implemented in PyTorch with LoRA support
  ☆676Updated last year
• KellerJordan / Muon
  Muon is an optimizer for hidden layers in neural networks
  ☆1,710Updated 2 months ago
• meta-pytorch / LeanRL
  LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
  ☆623Updated 3 weeks ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆238Updated last year
• apple / ml-tarflow
  ☆278Updated 9 months ago