flukeskywalker / nanoDDLinks
Simple Scalable Discrete Diffusion for text in PyTorch
☆37Updated last year
Alternatives and similar repositories for nanoDD
Users that are interested in nanoDD are comparing it to the libraries listed below
Sorting:
- ☆36Updated 10 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆42Updated last year
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Updated 2 years ago
- ☆32Updated last year
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆87Updated 2 years ago
- ☆53Updated last week
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆71Updated 10 months ago
- ☆42Updated 3 years ago
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated 2 years ago
- Transformers with doubly stochastic attention☆53Updated 3 years ago
- The Superposition of Diffusion Models Using the Itô Density Estimator☆51Updated 10 months ago
- Implementation of Action Matching☆51Updated 2 years ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆58Updated 2 years ago
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.☆68Updated 9 months ago
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆141Updated 2 years ago
- 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
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Updated 11 months ago
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago
- Official Jax Implementation of MD4 Masked Diffusion Models☆153Updated 11 months ago
- ☆62Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆66Updated last year
- ☆130Updated last year
- ☆37Updated last week
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆129Updated 6 months ago
- The Energy Transformer block, in JAX☆63Updated 2 years ago
- ☆111Updated 2 years ago
- A simple implimentation of Bayesian Flow Networks (BFN)☆240Updated 2 years ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆51Updated last year
- [ICLR 2022] Path integral sampler☆52Updated 2 years ago
- Code release for "Stochastic Optimal Control Matching"☆39Updated last year