Alternatives and similar repositories for smalldiffusion:

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

• tonyduan / diffusion
  From-scratch diffusion model implemented in PyTorch.
  ☆81Updated 11 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆261Updated 7 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆214Updated last week
• tanelp / tiny-diffusion
  A minimal PyTorch implementation of probabilistic diffusion models for 2D datasets.
  ☆696Updated 8 months ago
• helblazer811 / diffusion-visualizations
  Visualizations of the theory behind diffusion models.
  ☆77Updated 8 months ago
• KellerJordan / cifar10-airbench
  94% on CIFAR-10 in 2.6 seconds 💨 96% in 27 seconds
  ☆195Updated last month
• cloneofsimo / minRF
  Minimal implementation of scalable rectified flow transformers, based on SD3's approach
  ☆465Updated 6 months ago
• fferflo / einx
  Universal Tensor Operations in Einstein-Inspired Notation for Python.
  ☆341Updated last month
• cloneofsimo / d3pm
  Minimal Implementation of a D3PM in pytorch
  ☆192Updated 8 months ago
• VSehwag / minimal-diffusion
  A minimal yet resourceful implementation of diffusion models (along with pretrained models + synthetic images for nine datasets)
  ☆267Updated 4 months ago
• google-deepmind / nanodo
  ☆201Updated 6 months ago
• kvfrans / shortcut-models
  ☆320Updated last month
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated last year
• lucidrains / nGPT-pytorch
  Quick implementation of nGPT, learning entirely on the hypersphere, from NvidiaAI
  ☆270Updated 2 months ago
• buoyancy99 / diffusion-forcing
  code for "Diffusion Forcing: Next-token Prediction Meets Full-Sequence Diffusion"
  ☆684Updated last month
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆119Updated 5 months ago
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆252Updated 2 months ago
• facebookresearch / optimizers
  For optimization algorithm research and development.
  ☆484Updated this week
• pytorch-labs / LeanRL
  LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.
  ☆486Updated 2 months ago
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆144Updated 2 years ago
• cabralpinto / modular-diffusion
  Python library for designing and training your own Diffusion Models with PyTorch.
  ☆271Updated 5 months ago
• srush / annotated-mamba
  Annotated version of the Mamba paper
  ☆469Updated 10 months ago
• epistoteles / TensorHue
  TensorHue is a Python library that allows you to visualize tensors right in your console, making understanding and debugging tensor conte…
  ☆115Updated 3 months ago
• Kinyugo / consistency_models
  A mini-library for training consistency models.
  ☆234Updated last year
• wilson-labs / cola
  Compositional Linear Algebra
  ☆456Updated this week
• yklcs / jaxsplat
  3D Gaussian Splatting in JAX
  ☆55Updated 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)
  ☆454Updated 10 months ago
• martenlienen / torchode
  A parallel ODE solver for PyTorch
  ☆244Updated 3 months ago
• BlackHC / neural_net_checklist
  ☆149Updated 5 months ago
• gmongaras / Diffusion_models_from_scratch
  Creating a diffusion model from scratch in PyTorch to learn exactly how they work.
  ☆342Updated 8 months ago