Alternatives and similar repositories for nanoDD

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

• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆41Updated last year
• nkalyanv99 / UNI-D2
  ☆53Updated 3 weeks ago
• timgaripov / compositional-sculpting
  Code for paper "Compositional Sculpting of Iterative Generative Processes"
  ☆25Updated 2 years ago
• AndyShih12 / LongHorizonTemperatureScaling
  PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023
  ☆20Updated 2 years ago
• necludov / wl-mechanics
  ☆31Updated last year
• GFNOrg / gfn-diffusion
  ☆36Updated 9 months ago
• necludov / super-diffusion
  The Superposition of Diffusion Models Using the Itô Density Estimator
  ☆50Updated 9 months ago
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆51Updated 3 years ago
• ruqizhang / discrete-langevin
  ☆42Updated 3 years ago
• GFNOrg / EB_GFN
  Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"
  ☆87Updated 2 years ago
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆71Updated 10 months ago
• DoronHav / WassersteinFlowMatching
  ☆36Updated this week
• google-deepmind / md4
  Official Jax Implementation of MD4 Masked Diffusion Models
  ☆151Updated 10 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆51Updated last year
• tychovdo / noethers-razor
  Code for NeurIPS 2024 paper: "Noether's razor: Learning Conserved Quantities" by Tycho F. A. van der Ouderaa, Mark van der Wilk, Pim de H…
  ☆11Updated last year
• lazaratan / meta-flow-matching
  Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold
  ☆69Updated 10 months ago
• olsdavis / fisher-flow
  Official implementation of Fisher-Flow Matching (NeurIPS 2024).
  ☆35Updated last year
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆25Updated 2 years ago
• GallagherCommaJack / modulax
  ☆18Updated last year
• lindermanlab / elk
  Scalable and Stable Parallelization of Nonlinear RNNS
  ☆28Updated 2 months ago
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆63Updated 2 years ago
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆128Updated 5 months ago
• sooheon / perceiver-jax
  Perceiver (transformer variant) implemented in JAX and Flax
  ☆13Updated 4 years ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated last month
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆39Updated 3 years ago
• shikaiqiu / compute-better-spent
  ☆62Updated last year
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆34Updated 2 years ago
• liusulin / DDPD
  Official Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising" [ICLR 2025]
  ☆84Updated 8 months ago
• wgrathwohl / GWG_release
  Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"
  ☆58Updated 2 years ago
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Updated 2 years ago