Alternatives and similar repositories for tiny-tiny-diffusion

Users that are interested in tiny-tiny-diffusion are comparing it to the libraries listed below

• daigo0927 / jax-ddim
  Jax/Flax implementation of Denoising Diffusion Implicit Models
  ☆20Updated 3 years ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆104Updated last year
• lucidrains / hl-gauss-pytorch
  The Gaussian Histogram Loss (HL-Gauss) proposed by Imani et al. with a few convenient wrappers for regression, in Pytorch
  ☆65Updated last month
• lebellig / flow-matching
  Annotated Flow Matching paper
  ☆208Updated last year
• apple / ml-agm
  ☆44Updated last year
• SKholkin / LightSB-Matching
  Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation
  ☆48Updated last year
• lucidrains / gradnorm-pytorch
  A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch
  ☆108Updated last month
• enhuiz / langevin-dynamics-and-score-matching
  A demo shows how to combine Langevin dynamics with score matching for generative models.
  ☆40Updated 4 years ago
• Jhomanik / Optimal-Flow-Matching
  The official repository for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step" (NeurIPS 2024)
  ☆91Updated 9 months ago
• tatsy / normalizing-flows-pytorch
  PyTorch implementations of normalizing flow and its variants.
  ☆78Updated 4 years ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆36Updated last year
• kvfrans / splus
  ☆120Updated 3 months ago
• AaronHavens / FlowMatching
  A repo where I play with conditional flow approaches for learning time-varying vector-fields.
  ☆23Updated last year
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆126Updated 6 months ago
• apple / ml-tarflow
  ☆279Updated 9 months ago
• facebookresearch / generalized-schrodinger-bridge-matching
  code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).
  ☆79Updated last year
• lebellig / discrete-fm
  Educational implementation of the Discrete Flow Matching paper
  ☆116Updated last year
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆47Updated 3 years ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated last year
• amorehead / jvp_flash_attention
  Flash Attention Triton kernel with support for second-order derivatives
  ☆88Updated this week
• louaaron / Reflected-Diffusion
  [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)
  ☆158Updated last year
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆43Updated 2 years ago
• yilundu / reduce_reuse_recycle
  ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC
  ☆145Updated 11 months ago
• filipbasara0 / simple-diffusion
  A minimal implementation of a denoising diffusion model in PyTorch.
  ☆127Updated 10 months ago
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆58Updated last year
• lucidrains / adam-atan2-pytorch
  Implementation of the proposed Adam-atan2 from Google Deepmind in Pytorch
  ☆124Updated 10 months ago
• vdutor / neural-diffusion-processes
  Neural Diffusion Processes
  ☆81Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆42Updated 9 months ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆28Updated last year
• yuanzhi-zhu / mini_edm
  Minimum implementation of EDM (Elucidating the Design Space of Diffusion-Based Generative Models) on cifar10 and mnist
  ☆58Updated last year