michaelsdr / sinkformersLinks
Transformers with doubly stochastic attention
☆47Updated 2 years ago
Alternatives and similar repositories for sinkformers
Users that are interested in sinkformers are comparing it to the libraries listed below
Sorting:
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆125Updated last year
- Sequence Modeling with Structured State Spaces☆66Updated 3 years ago
- Code repository of the paper "CKConv: Continuous Kernel Convolution For Sequential Data" published at ICLR 2022. https://arxiv.org/abs/21…☆123Updated 2 years ago
- Neural Diffusion Processes☆81Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆42Updated last year
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Updated 6 months ago
- JAX exponential map normalising flows on sphere☆17Updated 4 years ago
- Pytorch code for "Improving Self-Supervised Learning by Characterizing Idealized Representations"☆41Updated 2 years ago
- Implementation of Gated State Spaces, from the paper "Long Range Language Modeling via Gated State Spaces", in Pytorch☆101Updated 2 years ago
- Repository for the "Gotta Go Fast When Generating Data with Score-Based Models" paper☆105Updated 3 years ago
- Stochastic Normalizing Flows☆78Updated 3 years ago
- Euclidean Wasserstein-2 optimal transportation☆47Updated 2 years ago
- ☆31Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆65Updated 5 months ago
- A minimalist implementation of score-based diffusion model☆129Updated 4 years ago
- Code for paper "Argmax Flows and Multinomial Diffusion: Learning Categorical Distributions"☆90Updated 4 years ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆24Updated last year
- ☆17Updated 2 years ago
- ☆38Updated 2 years ago
- General Invertible Transformations for Flow-based Generative Models☆18Updated 4 years ago
- ☆33Updated 2 years ago
- This repository contains code for applying Riemannian geometry in machine learning.☆77Updated 4 years ago
- ☆164Updated 2 years ago
- ☆63Updated last year
- Implementation of Action Matching☆44Updated 2 years ago
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆34Updated 3 years ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆42Updated 2 months ago
- Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)☆148Updated 3 years ago
- Implementation of Action Matching for the Schrödinger equation☆23Updated 2 years ago
- [ICML'21] Improved Contrastive Divergence Training of Energy Based Models☆63Updated 3 years ago