Alternatives and similar repositories for energy-transformer-graph

Users that are interested in energy-transformer-graph are comparing it to the libraries listed below

• lindermanlab / elk
  Scalable and Stable Parallelization of Nonlinear RNNS
  ☆16Updated 4 months ago
• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆58Updated last year
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆41Updated last year
• AndPotap / einsum-search
  ☆32Updated 8 months ago
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆52Updated 5 months ago
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆24Updated 2 years ago
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆78Updated last year
• Benjamin-Walker / selective-ssms-and-linear-cdes
  Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)
  ☆15Updated 5 months ago
• AhmedImtiazPrio / grok-adversarial
  Deep Networks Grok All the Time and Here is Why
  ☆37Updated last year
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆41Updated 6 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆69Updated 7 months ago
• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆40Updated last year
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆89Updated last year
• opallab / positional_attention
  Source code for the paper "Positional Attention: Expressivity and Learnability of Algorithmic Computation"
  ☆14Updated last month
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated 11 months ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆91Updated 2 weeks ago
• augustwester / gflownet
  A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)
  ☆42Updated last year
• IdoAmos / not-from-scratch
  ☆31Updated 8 months ago
• shikaiqiu / compute-better-spent
  ☆53Updated 8 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 5 months ago
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 6 months ago
• GFNOrg / gfn-diffusion
  ☆34Updated 2 months ago
• tychovdo / noethers-razor
  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
• necludov / wl-mechanics
  ☆32Updated last year
• aradha / deep_neural_feature_ansatz
  Code for verifying deep neural feature ansatz
  ☆19Updated 2 years ago
• patrick-kidger / esm2quinox
  An implementation of ESM2 in Equinox+JAX
  ☆29Updated 3 weeks ago
• timgaripov / compositional-sculpting
  Code for paper "Compositional Sculpting of Iterative Generative Processes"
  ☆22Updated last year
• mcbal / afem
  Implementation of approximate free-energy minimization in PyTorch
  ☆19Updated 3 years ago
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆33Updated 2 years ago