Baran-phys / Tropical-AttentionLinks
[NeurIPS 2025] Official code for "Tropical Attention: Neural Algorithmic Reasoning for Combinatorial Algorithms"
☆22Updated 2 months ago
Alternatives and similar repositories for Tropical-Attention
Users that are interested in Tropical-Attention are comparing it to the libraries listed below
Sorting:
- The Energy Transformer block, in JAX☆63Updated 2 years ago
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Updated last year
- Neural Optimal Transport with Lagrangian Costs☆61Updated 7 months ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆48Updated 2 years ago
- Deep Networks Grok All the Time and Here is Why☆38Updated last year
- ☆33Updated last year
- ☆35Updated last year
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆92Updated last year
- ☆13Updated 2 years ago
- Official implementation of E(n)-equivariant Graph Neural Cellular Automata☆35Updated 3 months ago
- Parallelizing non-linear sequential models over the sequence length☆56Updated 6 months ago
- ☆35Updated last year
- Multi-Agent Verification: Scaling Test-Time Compute with Multiple Verifiers☆25Updated 10 months ago
- Semi-Supervised Offline Reinforcement Learning with Action-Free Trajectories☆42Updated 2 years ago
- Code for Lie Symmetries SSL paper☆24Updated last year
- ☆16Updated last year
- Official repository for the paper "Neural Differential Equations for Learning to Program Neural Nets Through Continuous Learning Rules" (…☆23Updated 7 months ago
- A set of tests for evaluating large-scale algorithms for Wasserstein-1 transport computation (NeurIPS'22).☆21Updated last year
- This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classificatio…☆25Updated last year
- A library implementing the kernels for and experiments using extrinsic gauge equivariant vector field Gaussian Processes☆25Updated 4 years ago
- Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group☆37Updated last year
- Lottery Tickets in Evolutionary Optimization (Lange & Sprekeler, ICML 2023)☆17Updated 2 years ago
- ☆63Updated 3 months ago
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14Updated 7 months ago
- Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…☆25Updated last year
- ☆36Updated 9 months ago
- Geometric Dynamic Variational Autoencoders (GD-VAEs) for learning embedding maps for nonlinear dynamics into general latent spaces. This …☆35Updated 2 months ago
- Pytorch (PyG) and Tensorflow (Keras/Spektral) implementation of Total Variation Graph Neural Network (TVGNN), as presented at ICML 2023.☆20Updated 10 months ago
- Meta-learning inductive biases in the form of useful conserved quantities.☆39Updated 3 years ago