Alternatives and similar repositories for smalldiffusion:

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

• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆272Updated 10 months ago
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆516Updated 9 months ago
• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆187Updated last year
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆730Updated 11 months ago
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆825Updated 3 weeks ago
• tonyduan / diffusion
  From-scratch diffusion model implemented in PyTorch.
  ☆87Updated last year
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆508Updated this week
• facebookresearch / flow_matching
  A PyTorch library for implementing flow matching algorithms, featuring continuous and discrete flow matching implementations. It includes…
  ☆2,425Updated 3 months ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆483Updated last year
• clu0 / unet.cu
  UNet diffusion model in pure CUDA
  ☆602Updated 9 months ago
• SonyResearch / micro_diffusion
  Official repository for our work on micro-budget training of large-scale diffusion models.
  ☆1,393Updated 3 months ago
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆275Updated 6 months ago
• facebookresearch / jepa-intuitive-physics
  This repo contains the code for the paper "Intuitive physics understanding emerges fromself-supervised pretraining on natural videos"
  ☆142Updated 2 months ago
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆279Updated last month
• KellerJordan / cifar10-airbench
  CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds
  ☆230Updated last month
• iShohei220 / adopt
  Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"
  ☆425Updated 4 months ago
• lumalabs / imm
  Official implementation of Inductive Moment Matching
  ☆451Updated last month
• pytorch-labs / LeanRL
  LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
  ☆547Updated 6 months ago
• kvfrans / shortcut-models
  ☆457Updated 4 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated last year
• gmongaras / Diffusion_models_from_scratch
  Creating a diffusion model from scratch in PyTorch to learn exactly how they work.
  ☆358Updated 3 weeks ago
• apapiu / transformer_latent_diffusion
  Text to Image Latent Diffusion using a Transformer core
  ☆178Updated 7 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)
  ☆556Updated last year
• NVlabs / edm2
  EDM2 and Autoguidance -- Official PyTorch implementation
  ☆693Updated 4 months ago
• KellerJordan / Muon
  Muon optimizer: +>30% sample efficiency with <3% wallclock overhead
  ☆577Updated last month
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆215Updated last year
• facebookresearch / schedule_free
  Schedule-Free Optimization in PyTorch
  ☆2,146Updated 2 weeks ago
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆182Updated 3 weeks ago
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆367Updated 2 weeks ago
• VSehwag / minimal-diffusion
  A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)
  ☆282Updated 7 months ago