Alternatives and similar repositories for smalldiffusion:

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

• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆148Updated 11 months ago
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆715Updated 10 months ago
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆489Updated 8 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆269Updated 9 months ago
• facebookresearch / flow_matching
  A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…
  ☆2,206Updated 2 months ago
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆219Updated 3 weeks ago
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆276Updated this week
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆270Updated 5 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated last year
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆498Updated this week
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆777Updated last week
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆600Updated 8 months ago
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆205Updated 11 months ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆475Updated last year
• minyoungg / platonic-rep
  ☆505Updated 7 months ago
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆659Updated 3 months ago
• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆172Updated 6 months ago
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,116Updated 3 weeks ago
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆419Updated 3 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)
  ☆530Updated last year
• pytorch-labs / LeanRL
  LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
  ☆524Updated 4 months ago
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆361Updated last month
• kvfrans / shortcut-models
  ☆411Updated 3 months ago
• KellerJordan / Muon
  Muon optimizer: +>30% sample efficiency with <3% wallclock overhead
  ☆505Updated last week
• SonyResearch / micro_diffusion
  Official repository for our work on micro-budget training of large-scale diffusion models.
  ☆1,356Updated 2 months ago
• tonyduan / diffusion
  From-scratch diffusion model implemented in PyTorch.
  ☆83Updated last year
• google-deepmind / nanodo
  ☆214Updated 8 months ago
• BlackHC / neural_net_checklist
  ☆149Updated 7 months ago