michaelsdr / sinkformers
Transformers with doubly stochastic attention
☆45Updated 2 years ago
Alternatives and similar repositories for sinkformers:
Users that are interested in sinkformers are comparing it to the libraries listed below
- Implementation of Action Matching☆38Updated last year
- Sequence Modeling with Structured State Spaces☆63Updated 2 years ago
- Euclidean Wasserstein-2 optimal transportation☆45Updated last year
- Stochastic Normalizing Flows☆76Updated 3 years ago
- ☆32Updated 9 months ago
- Neural Diffusion Processes☆76Updated 6 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆122Updated last year
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Official Jax Implementation of MD4 Masked Diffusion Models☆61Updated this week
- ☆17Updated last year
- Code release for "Stochastic Optimal Control Matching"☆30Updated 6 months ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆59Updated 6 months ago
- ☆53Updated 7 months 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 code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated 2 years ago
- Code for the paper https://arxiv.org/abs/2205.14987v2☆48Updated 10 months ago
- NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models☆14Updated 4 months ago