Transformers with doubly stochastic attention
☆54Sep 14, 2022Updated 3 years ago
Alternatives and similar repositories for sinkformers
Users that are interested in sinkformers are comparing it to the libraries listed below
Sorting:
- Learnable Global Pooling Layers Based on Regularized Optimal Transport (ROT)☆16Mar 17, 2024Updated last year
- [ICML2025] The official implementation of "WGFormer: An SE(3)-Transformer Driven by Wasserstein Gradient Flows for Molecular Ground-State…☆32Jun 7, 2025Updated 9 months ago
- A list of papers for group meeting☆19Jan 5, 2026Updated 2 months ago
- [ICML2025] The official implementation of "PolyConf: Unlocking Polymer Conformation Generation through Hierarchical Generative Models"☆34Jan 30, 2026Updated last month
- ☆10Jun 14, 2023Updated 2 years ago
- ☆24Sep 10, 2021Updated 4 years ago
- ☆11Apr 10, 2023Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 2 years ago
- Pragmatic models for generating and following instructions☆13Dec 22, 2019Updated 6 years ago
- orbital MCMC☆10Jun 17, 2021Updated 4 years ago
- This is the code related to the article 'Intrinsic persistent homology via density-based metric learning'☆10May 8, 2023Updated 2 years ago
- Graph matching and clustering by comparing heat kernels via optimal transport.☆27Dec 8, 2022Updated 3 years ago
- [WIP] PyTorch & Hydra Template☆12Jul 13, 2022Updated 3 years ago
- Python package for the ICML 2022 paper "Unsupervised Ground Metric Learning Using Wasserstein Singular Vectors".☆10Sep 2, 2024Updated last year
- Geometric Algebra Flow Matching (GAFL) for Protein Backbone Generation☆17Oct 31, 2025Updated 4 months ago
- ☆13Sep 13, 2023Updated 2 years ago
- ☆18Sep 18, 2024Updated last year
- This repository contains code for the paper: S Bergsma, T Zeyl, JR Anaraki, L Guo, C2FAR: Coarse-to-Fine Autoregressive Networks for Prec…☆13Dec 7, 2023Updated 2 years ago
- ☆33Dec 17, 2025Updated 2 months ago
- ☆12Jul 25, 2024Updated last year
- PyTorch Tutorial at the LOD2021 conference☆21Oct 7, 2021Updated 4 years ago
- Python implementation of the supervised graph prediction method proposed in http://arxiv.org/abs/2202.03813 using PyTorch library and POT…☆16Feb 25, 2022Updated 4 years ago
- Code for computing Hypergraph Co-Optimal Transport distances☆17Jan 19, 2023Updated 3 years ago
- Source code for Noise-Contrastive Estimation for Multivariate Point Processes (NeurIPS 2020).☆15Nov 3, 2020Updated 5 years ago
- Code for ECML/PKDD paper: "LSMI-Sinkhorn: Semi-supervised Mutual Information Estimation with Optimal Transport"☆17Jun 27, 2021Updated 4 years ago
- Domain Agnostic Fourier Neural Operators (DAFNO)☆19Sep 3, 2024Updated last year
- Official code for Fisher information embedding for node and graph learning (ICML 2023)☆19Jun 17, 2023Updated 2 years ago
- CO-Optimal Transport☆42Nov 6, 2020Updated 5 years ago
- ☆18Jun 26, 2023Updated 2 years ago
- Approximate subgraph motif mining through learnable edge contraction.☆15Aug 3, 2023Updated 2 years ago
- ☆19May 25, 2023Updated 2 years ago
- ☆18Jul 19, 2022Updated 3 years ago
- Code for http://proceedings.mlr.press/v80/dvurechensky18a.html☆17Jul 31, 2018Updated 7 years ago
- A pip installable package for optimal transport inspired loss functions in the spectral domain. Can be used for audio applications such a…☆29Dec 5, 2025Updated 3 months ago
- Learning the optimal transport map via input convex neural neworks☆42Jun 26, 2020Updated 5 years ago
- Euclidean Wasserstein-2 optimal transportation☆46Aug 19, 2023Updated 2 years ago
- Computationally friendly hyper-parameter search with DP-SGD☆25Jan 7, 2025Updated last year
- [AISTATS2020] The official repository of "Invertible Generative Modling using Linear Rational Splines (LRS)".☆20May 12, 2023Updated 2 years ago
- Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload☆132Aug 6, 2022Updated 3 years ago