michaelsdr / sinkformersLinks
Transformers with doubly stochastic attention
☆51Updated 3 years ago
Alternatives and similar repositories for sinkformers
Users that are interested in sinkformers are comparing it to the libraries listed below
Sorting:
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Updated 10 months ago
- ☆31Updated last year
- Neural Diffusion Processes☆81Updated last year
- Stochastic Normalizing Flows☆78Updated 4 years ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆127Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- This repository contains code for applying Riemannian geometry in machine learning.☆78Updated 4 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆70Updated 9 months ago
- Sequence Modeling with Structured State Spaces☆67Updated 3 years ago
- Official release of code for "Oops I Took A Gradient: Scalable Sampling for Discrete Distributions"☆57Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Updated 2 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- Code repository of the paper "CKConv: Continuous Kernel Convolution For Sequential Data" published at ICLR 2022. https://arxiv.org/abs/21…☆124Updated 3 years ago
- JAX exponential map normalising flows on sphere☆17Updated 5 years ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆26Updated 2 years ago
- A minimalist implementation of score-based diffusion model☆129Updated 4 years ago
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆105Updated 4 years ago
- ☆18Updated 2 years ago
- ☆64Updated last year
- Code for the paper https://arxiv.org/abs/2205.14987v2☆57Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆66Updated last year
- Implementation of Action Matching☆49Updated 2 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 3 years ago
- Score-Based Generative Modeling with Critically-Damped Langevin Diffusion☆200Updated 2 years ago
- ☆35Updated this week
- Simple Scalable Discrete Diffusion for text in PyTorch☆37Updated last year
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆69Updated 3 years ago
- PyTorch implementation of the OT-Flow approach in arXiv:2006.00104☆57Updated last year
- Modern Fixed Point Systems using Pytorch☆125Updated 2 years ago
- ☆39Updated 3 years ago