Lemon-cmd / energy-transformer-graphLinks
This repository contains the official code for Energy Transformer---an efficient Energy-based Transformer variant for graph classification
☆24Updated 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:
- Scalable and Stable Parallelization of Nonlinear RNNS☆16Updated 4 months ago
- The Energy Transformer block, in JAX☆58Updated last year
- Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX☆41Updated last year
- ☆32Updated 8 months ago
- Parallelizing non-linear sequential models over the sequence length☆52Updated 5 months ago
- Implementation of Action Matching for the Schrödinger equation☆24Updated 2 years ago
- A State-Space Model with Rational Transfer Function Representation.☆78Updated last year
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Updated 5 months ago
- Deep Networks Grok All the Time and Here is Why☆37Updated last year
- Lightning-like training API for JAX with Flax☆41Updated 6 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆83Updated last year
- Code repository for Trajectory Flow Matching☆69Updated 7 months ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆40Updated last year
- Implementation of GateLoop Transformer in Pytorch and Jax☆89Updated last year
- Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"☆14Updated last month
- Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network☆50Updated 11 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆91Updated 2 weeks ago
- A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)☆42Updated last year
- ☆31Updated 8 months ago
- ☆53Updated 8 months ago
- Implementation of PSGD optimizer in JAX☆33Updated 5 months ago
- Turn jitted jax functions back into python source code☆22Updated 6 months ago
- ☆34Updated 2 months ago
- Code for NeurIPS 2024 paper: "Noether's razor: Learning Conserved Quantities" by Tycho F. A. van der Ouderaa, Mark van der Wilk, Pim de H…☆10Updated 8 months ago
- ☆32Updated last year
- Code for verifying deep neural feature ansatz☆19Updated 2 years ago
- An implementation of ESM2 in Equinox+JAX☆29Updated 3 weeks ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆22Updated last year
- Implementation of approximate free-energy minimization in PyTorch☆19Updated 3 years ago
- An annotated implementation of the Hyena Hierarchy paper☆33Updated 2 years ago