flukeskywalker / nanoDD
Simple Scalable Discrete Diffusion for text in PyTorch
☆33Updated 5 months ago
Alternatives and similar repositories for nanoDD:
Users that are interested in nanoDD are comparing it to the libraries listed below
- Official Jax Implementation of MD4 Masked Diffusion Models☆67Updated 3 weeks ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated last year
- ☆24Updated last week
- The Superposition of Diffusion Models Using the Itô Density Estimator☆33Updated last month
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- ☆32Updated 9 months ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 2 months ago
- simple bibtex generator for any text with \cite{}☆31Updated 8 months ago
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆48Updated 2 weeks ago
- Derivative-Free Guidance in Diffusion Models with Soft Value-Based Decoding. For controlled generation in DNA, RNA, proteins, molecules (…☆28Updated 5 months ago
- ☆102Updated 10 months ago
- ☆52Updated 5 months ago
- Code for the paper https://arxiv.org/abs/2402.04997☆72Updated last year
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆20Updated last year
- Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"☆14Updated last month
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆49Updated 7 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆23Updated 2 months ago
- Flow-matching algorithms in JAX☆86Updated 7 months ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆82Updated 2 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated last week
- Simple Guidance Mechanisms for Discrete Diffusion Models☆29Updated 3 months ago
- ☆23Updated 3 months ago
- A simple DIffusion LM approach.☆22Updated last year
- ☆91Updated last year
- The Energy Transformer block, in JAX☆56Updated last year
- ☆30Updated 5 months ago