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☆59Updated last year
- ☆9Updated 2 years ago
- Parallelizing non-linear sequential models over the sequence length☆53Updated last month
- ☆32Updated 10 months ago
- Scalable and Stable Parallelization of Nonlinear RNNS☆17Updated 6 months ago
- A State-Space Model with Rational Transfer Function Representation.☆79Updated last year
- Code repository for Trajectory Flow Matching☆74Updated 9 months ago
- Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"☆85Updated 2 years ago
- A MAD laboratory to improve AI architecture designs 🧪☆123Updated 7 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
- ☆35Updated 4 months ago
- Explorations into whether a transformer with RL can direct a genetic algorithm to converge faster☆70Updated 2 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆85Updated last year
- Official implementation of Fisher-Flow Matching (NeurIPS 2024).☆26Updated 9 months ago
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Updated 7 months ago
- ☆53Updated 10 months ago
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆41Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆131Updated last year
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14Updated 2 months ago
- An annotated implementation of the Hyena Hierarchy paper☆33Updated 2 years ago
- Generative Flow Networks - GFlowNet☆267Updated 3 weeks ago
- ☆206Updated 8 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆95Updated last month
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆98Updated last month
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆115Updated 8 months ago
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated last year
- ☆32Updated last year
- Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.☆19Updated last year
- Lightning-like training API for JAX with Flax☆42Updated 8 months ago
- ☆33Updated 9 months ago