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☆40Updated last year
- ☆32Updated 10 months ago
- ☆17Updated last year
- Euclidean Wasserstein-2 optimal transportation☆46Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated 3 weeks ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Stochastic Normalizing Flows☆76Updated 3 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 for the paper https://arxiv.org/abs/2205.14987v2☆48Updated 11 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆43Updated last year
- Official repository of Implicit Neural Convolutional Kernels for Steerable CNNs, Zhdanov et al.☆28Updated last month
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆40Updated 2 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆62Updated 2 years ago
- ☆34Updated 2 years ago
- ☆17Updated 3 years ago
- ☆24Updated 3 years ago
- ☆33Updated 2 years ago
- Code release for "Stochastic Optimal Control Matching"☆32Updated 7 months ago
- Kernel Stein Discrepancy Descent : a method to sample from unnormalized densities☆21Updated 11 months ago
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆104Updated 3 years ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- This repository contains code for applying Riemannian geometry in machine learning.☆77Updated 3 years ago
- A minimalist implementation of score-based diffusion model☆123Updated 3 years ago
- ☆64Updated last year
- ☆53Updated 8 months ago
- ☆25Updated last week
- ☆19Updated 2 months ago
- [NeurIPS'20] Code for the Paper Compositional Visual Generation and Inference with Energy Based Models☆44Updated 2 years ago
- Code for PAC-Bayes Compression Bounds So Tight That They Can Explain Generalization, NeurIPS 2022☆15Updated 2 years ago
- Sequence Modeling with Structured State Spaces☆63Updated 2 years ago