Lemon-cmd / energy-transformer-graphLinks
This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classification
☆25Updated 2 years ago
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:
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆46Updated 2 years ago
- The Energy Transformer block, in JAX☆63Updated 2 years ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆87Updated 2 years ago
- Code repository for Trajectory Flow Matching☆100Updated last year
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆128Updated last year
- ☆62Updated last year
- ☆33Updated last year
- Generative Flow Networks - GFlowNet☆312Updated this week
- Scalable and Stable Parallelization of Nonlinear RNNS☆28Updated 3 months ago
- GflowNets, MCMC, Metropolis-Hasting, Gibbs sampling, Metropolis-adjusted Langevin, Inverse Transform Sampling, Acceptance-Rejection Metho…☆86Updated 2 years ago
- ☆36Updated 10 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆42Updated 2 years ago
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14Updated 8 months ago
- A State-Space Model with Rational Transfer Function Representation.☆83Updated last year
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆51Updated last year
- Parallelizing non-linear sequential models over the sequence length☆56Updated 7 months ago
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆111Updated 4 months ago
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆44Updated 2 years ago
- Omnigrok: Grokking Beyond Algorithmic Data☆62Updated 2 years ago
- ☆69Updated 10 months ago
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Updated last year
- DISCS: The code base for the Benchmark for Discrete Sampling☆63Updated last year
- ☆66Updated 4 months ago
- [ICML2025 Oral] LLM-SRBench: A New Benchmark for Scientific Equation Discovery with Large Language Models☆94Updated 6 months ago
- Implementation of approximate free-energy minimization in PyTorch☆21Updated 4 years ago
- DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule☆64Updated 2 years ago
- ☆28Updated 2 years ago
- ☆32Updated last year
- A MAD laboratory to improve AI architecture designs 🧪☆138Updated last year
- Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).☆82Updated last year