flukeskywalker / nanoDD
Simple Scalable Discrete Diffusion for text in PyTorch
☆32Updated 5 months ago
Alternatives and similar repositories for nanoDD:
Users that are interested in nanoDD are comparing it to the libraries listed below
- PyTorch implementation for "Long Horizon Temperature Scaling", ICML 2023☆20Updated last year
- Official Jax Implementation of MD4 Masked Diffusion Models☆64Updated 2 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
- ☆24Updated 5 months ago
- ☆52Updated 5 months ago
- Your favourite classical machine learning algos on the GPU/TPU☆20Updated 2 months ago
- Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch☆23Updated last month
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- Derivative-Free Guidance in Diffusion Models with Soft Value-Based Decoding. For controlled generation in DNA, RNA, proteins, molecules (…☆28Updated 5 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆49Updated 7 months ago
- ☆32Updated 9 months ago
- The Superposition of Diffusion Models Using the Itô Density Estimator☆28Updated 3 weeks ago
- An annotated implementation of the Hyena Hierarchy paper☆32Updated last year
- Code for minimum-entropy coupling.☆31Updated 8 months ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆20Updated last year
- ☆30Updated 5 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆87Updated 8 months ago
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆27Updated 2 weeks ago
- Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"☆14Updated last month
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆82Updated 2 years ago
- ☆34Updated 2 years ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆25Updated 9 months ago
- The Energy Transformer block, in JAX☆56Updated last year
- simple bibtex generator for any text with \cite{}☆31Updated 8 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆37Updated 2 years ago
- ☆21Updated 5 months ago
- ☆17Updated last year
- Euclidean Wasserstein-2 optimal transportation☆45Updated last year