Lemon-cmd / energy-transformer-graph

Related projects ⓘ

Alternatives and complementary repositories for energy-transformer-graph

• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆53Updated 11 months ago
• opallab / positional_attention
  Source code for the paper "Positional Attention: Out-of-Distribution Generalization and Expressivity for Neural Algorithmic Reasoning"
  ☆14Updated this week
• GFNOrg / GFlowNet-EM
  Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.
  ☆38Updated 9 months ago
• mkofinas / neural-graphs
  Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).
  ☆70Updated 3 months ago
• shikaiqiu / compute-better-spent
  ☆46Updated last month
• timgaripov / compositional-sculpting
  Code for paper "Compositional Sculpting of Iterative Generative Processes"
  ☆20Updated last year
• necludov / wl-mechanics
  ☆31Updated 5 months ago
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆70Updated 6 months ago
• augustwester / gflownet
  A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)
  ☆40Updated last year
• necludov / jam
  Implementation of Action Matching
  ☆36Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆79Updated 9 months ago
• GFNOrg / GFN_vs_HVI
  ☆10Updated last year
• thjashin / multires-conv
  Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)
  ☆120Updated last year
• mcbal / afem
  Implementation of approximate free-energy minimization in PyTorch
  ☆18Updated 3 years ago
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆31Updated last year
• flukeskywalker / nanoDD
  Simple Scalable Discrete Diffusion for text in PyTorch
  ☆28Updated last month
• ml-jku / DIffUCO
  ☆30Updated 2 weeks ago
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆45Updated 3 weeks ago
• necludov / action-matching
  Implementation of Action Matching for the Schrödinger equation
  ☆22Updated last year
• aradha / deep_neural_feature_ansatz
  Code for verifying deep neural feature ansatz
  ☆15Updated last year
• nZhangx / TrajectoryFlowMatching
  Code repository for Trajectory Flow Matching
  ☆24Updated 3 weeks ago
• J-zin / energy-discrepancy
  NeurIPS'23: Energy Discrepancies: A Score-Independent Loss for Energy-Based Models
  ☆13Updated 3 weeks ago
• take-koshizuka / NLSB
  ☆18Updated last year
• SamsungSAILMontreal / AnyMolGenCritic
  ☆20Updated last month
• TianrongChen / DMSB
  Official implementation of Deep Momentum Schrödinger Bridge
  ☆20Updated last year
• GFNOrg / EB_GFN
  Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"
  ☆76Updated last year
• facebookresearch / w2ot
  Euclidean Wasserstein-2 optimal transportation
  ☆44Updated last year
• liusulin / DDPD
  Code for Paper "Think While You Generate: Discrete Diffusion with Planned Denoising"
  ☆23Updated last week
• MaximeRobeyns / bayesian_flow_networks
  A PyTorch implementation of Bayesian flow networks (Graves et al., 2023).
  ☆23Updated 11 months ago
• GFNOrg / gfn-diffusion
  ☆20Updated last month