Alternatives and similar repositories for diffusion-visualizations:

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

• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆89Updated 8 months ago
• kvfrans / jax-diffusion-transformer
  Implementation of Diffusion Transformer (DiT) in JAX
  ☆272Updated 10 months ago
• smsharma / minified-generative-models
  Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…
  ☆128Updated last year
• BlackHC / neural_net_checklist
  ☆150Updated 8 months ago
• facebookresearch / riemannian-fm
  code for "Riemannian Flow Matching on General Geometries".
  ☆219Updated last year
• yuanchenyang / smalldiffusion
  Simple and readable code for training and sampling from diffusion models
  ☆478Updated 3 months ago
• dome272 / Flow-Matching
  My take on Flow Matching
  ☆52Updated 3 months ago
• Sohl-Dickstein / fractal
  The boundary of neural network trainability is fractal
  ☆198Updated last year
• misko / human_descent
  ☆36Updated 4 months ago
• Algomancer / Bayesian-Flow-Networks
  A simple implimentation of Bayesian Flow Networks (BFN)
  ☆240Updated last year
• apoorvkh / academic-pretraining
  $100K or 100 Days: Trade-offs when Pre-Training with Academic Resources
  ☆135Updated last month
• ml-jku / GINNs-Geometry-informed-Neural-Networks
  ☆161Updated 2 months ago
• lebellig / flow-matching
  Annotated Flow Matching paper
  ☆181Updated 7 months ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆73Updated 2 months ago
• wilson-labs / cola
  Compositional Linear Algebra
  ☆473Updated 3 weeks ago
• KindXiaoming / grow-crystals
  Getting crystal-like representations with harmonic loss
  ☆182Updated 3 weeks ago
• eje24 / iap-diffusion-labs
  Labs for MIT 6.S184/6.S975, IAP 2025
  ☆67Updated last month
• nikhilvyas / SOAP
  ☆175Updated 4 months ago
• malbergo / stochastic-interpolants
  ☆141Updated last year
• riverstone496 / awesome-second-order-optimization
  ☆25Updated last year
• modula-systems / modula
  🧱 Modula software package
  ☆188Updated 3 weeks ago
• kuleshov-group / awesome-discrete-diffusion-models
  A curated list for awesome discrete diffusion models resources.
  ☆299Updated 2 months ago
• zacharyhorvitz / Fk-Diffusion-Steering
  A general framework for inference-time scaling and steering of diffusion models with arbitrary rewards.
  ☆127Updated 2 months ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆101Updated 3 weeks ago
• david-knigge / enf-min-jax
  A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).
  ☆23Updated 2 months ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆236Updated 2 months ago
• francoisfleuret / picogpt
  Minimal GPT (~350 lines with a simple task to test it)
  ☆62Updated 4 months ago
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆239Updated last year
• tonyduan / diffusion
  From-scratch diffusion model implemented in PyTorch.
  ☆87Updated last year
• cloneofsimo / vqgan-training
  Train VAE like a boss
  ☆275Updated 6 months ago