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☆58Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆82Updated last year
- Parallelizing non-linear sequential models over the sequence length☆54Updated 4 months ago
- ☆33Updated last year
- Code repository for Trajectory Flow Matching☆88Updated 11 months ago
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14Updated 5 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆41Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆131Updated 10 months ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆104Updated 2 weeks ago
- Scalable and Stable Parallelization of Nonlinear RNNS☆23Updated last week
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆88Updated last year
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆83Updated last year
- An annotated implementation of the Hyena Hierarchy paper☆34Updated 2 years ago
- ☆66Updated 7 months ago
- ☆31Updated last year
- Lightning-like training API for JAX with Flax☆44Updated 10 months ago
- ☆33Updated 7 months ago
- ☆58Updated last year
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆104Updated last week
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆84Updated 2 years ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- GflowNets, MCMC, Metropolis-Hasting, Gibbs sampling, Metropolis-adjusted Langevin, Inverse Transform Sampling, Acceptance-Rejection Metho…☆86Updated 2 years ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆43Updated last year
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆63Updated 2 years ago
- Generative Flow Networks - GFlowNet☆286Updated 2 weeks ago
- Official implementation of Fisher-Flow Matching (NeurIPS 2024).☆33Updated last year
- Omnigrok: Grokking Beyond Algorithmic Data☆62Updated 2 years ago
- Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.☆21Updated last year
- ☆34Updated last year
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆43Updated 2 years ago