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.
  ☆730Updated last week
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆564Updated 11 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆276Updated 11 months ago
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆876Updated 2 months ago
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆741Updated last year
• tonyduan / diffusion
  From-scratch diffusion model implemented in PyTorch.
  ☆92Updated last year
• SonyResearch / micro_diffusion
  Official repository for our work on micro-budget training of large-scale diffusion models.
  ☆1,420Updated 4 months ago
• lumalabs / imm
  Official implementation of Inductive Moment Matching
  ☆475Updated 2 months ago
• facebookresearch / flow_matching
  A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…
  ☆2,699Updated last week
• kvfrans / shortcut-models
  ☆530Updated 6 months ago
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆719Updated 5 months ago
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆279Updated 7 months ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆482Updated last year
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆374Updated last month
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆606Updated 11 months ago
• louaaron / Score-Entropy-Discrete-Diffusion
  [ICML 2024 Best Paper] Discrete Diffusion Modeling by Estimating the Ratios of the Data Distribution (https://arxiv.org/abs/2310.16834)
  ☆581Updated last year
• kuleshov-group / awesome-discrete-diffusion-models
  A curated list for awesome discrete diffusion models resources.
  ☆349Updated this week
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆423Updated 5 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated last year
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,162Updated 2 weeks ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆518Updated this week
• lebellig / flow-matching
  Annotated Flow Matching paper
  ☆187Updated 8 months ago
• google-deepmind / nanodo
  ☆267Updated 10 months ago
• Kinyugo / consistency_models
  A mini-library for training consistency models.
  ☆244Updated last year
• pytorch / tensordict
  TensorDict is a pytorch dedicated tensor container.
  ☆925Updated last week
• dongzhuoyao / awesome-flow-matching
  A summary of related works about flow matching, stochastic interpolants
  ☆466Updated 2 months ago
• minyoungg / platonic-rep
  ☆556Updated last month
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆226Updated last year
• lucidrains / autoregressive-diffusion-pytorch
  Implementation of Autoregressive Diffusion in Pytorch
  ☆382Updated 7 months ago
• pytorch-labs / LeanRL
  LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
  ☆578Updated 7 months ago