Lemon-cmd / energy-transformer-graph

Related projects: ⓘ

• bhoov / energy-transformer-jax
  The Energy Transformer block, in JAX
  ☆48Updated 9 months ago
• shikaiqiu / compute-better-spent
  ☆42Updated 3 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.
  ☆38Updated 7 months ago
• wiseodd / lapeft-bayesopt
  Discrete Bayesian optimization with LLMs, PEFT finetuning methods, and the Laplace approximation.
  ☆14Updated last month
• timgaripov / compositional-sculpting
  Code for paper "Compositional Sculpting of Iterative Generative Processes"
  ☆20Updated 11 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆74Updated 7 months ago
• mcbal / spin-model-transformers
  Physics-inspired transformer modules based on mean-field dynamics of vector-spin models in JAX
  ☆32Updated 9 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆84Updated 4 months ago
• necludov / wl-mechanics
  ☆30Updated 3 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆28Updated 4 months ago
• LiibanMo / scikit-jax
  Your favourite classical machine learning algos on the GPU/TPU
  ☆20Updated this week
• facebookresearch / LAWT
  Code for papers Linear Algebra with Transformers (TMLR) and What is my Math Transformer Doing? (AI for Maths Workshop, Neurips 2022)
  ☆61Updated last month
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆86Updated 3 months ago
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆61Updated 4 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆62Updated last month
• mcbal / afem
  Implementation of approximate free-energy minimization in PyTorch
  ☆17Updated 2 years ago
• augustwester / gflownet
  A PyTorch implementation of a Generative Flow Network (GFlowNet) proposed by Bengio et al. (2021)
  ☆40Updated last year
• juliusberner / sde_sampler
  Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
  ☆48Updated 3 weeks ago
• GFNOrg / EB_GFN
  Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"
  ☆73Updated last year
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆45Updated last month
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year
• GallagherCommaJack / modulax
  ☆14Updated 3 weeks ago
• formll / dog
  DoG is SGD's Best Friend: A Parameter-Free Dynamic Step Size Schedule
  ☆57Updated last year
• radarFudan / mamba-minimal-jax
  ☆28Updated last week
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders in JAX.
  ☆52Updated last month
• epfml / schedules-and-scaling
  ☆47Updated 3 months ago
• Leiay / looped_transformer
  ☆18Updated 9 months ago
• jarridrb / DEM
  Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.
  ☆38Updated 7 months ago
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆40Updated last month
• mkofinas / neural-graphs
  Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).
  ☆63Updated last month