Lemon-cmd / energy-transformer-graphLinks
This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classification
☆25Updated last year
Alternatives and similar repositories for energy-transformer-graph
Users that are interested in energy-transformer-graph are comparing it to the libraries listed below
Sorting:
- The Energy Transformer block, in JAX☆61Updated last year
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14Updated 5 months ago
- ☆61Updated last year
- ☆33Updated last year
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆84Updated 2 years ago
- Code repository for Trajectory Flow Matching☆89Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆133Updated 11 months ago
- A State-Space Model with Rational Transfer Function Representation.☆83Updated 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
- Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.☆21Updated last year
- Parallelizing non-linear sequential models over the sequence length☆55Updated 4 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆105Updated last month
- Scalable and Stable Parallelization of Nonlinear RNNS☆25Updated 3 weeks ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆51Updated last year
- ☆34Updated 7 months ago
- ☆31Updated last year
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆63Updated 2 years ago
- ☆66Updated 7 months ago
- Implementation of GateLoop Transformer in Pytorch and Jax☆90Updated last year
- ☆33Updated last year
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆43Updated last year
- Generative Flow Networks - GFlowNet☆292Updated this week
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆83Updated last year
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Code for verifying deep neural feature ansatz☆20Updated 2 years ago
- GflowNets, MCMC, Metropolis-Hasting, Gibbs sampling, Metropolis-adjusted Langevin, Inverse Transform Sampling, Acceptance-Rejection Metho…☆86Updated 2 years ago
- Simple Scalable Discrete Diffusion for text in PyTorch☆37Updated last year
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Updated 10 months ago
- Repository for code used in the xVal paper☆144Updated last year
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆105Updated 3 weeks ago