facebookresearch / LAWT

Related projects: ⓘ

• yilundu / irem_code_release
  ICML 2022: Learning Iterative Reasoning through Energy Minimization
  ☆42Updated last year
• shikaiqiu / compute-better-spent
  ☆42Updated 3 months ago
• athms / mad-lab
  A MAD laboratory to improve AI architecture designs 🧪
  ☆84Updated 4 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆28Updated 4 months ago
• mkofinas / neural-graphs
  Official source code for "Graph Neural Networks for Learning Equivariant Representations of Neural Networks". In ICLR 2024 (oral).
  ☆63Updated last month
• lucidrains / coordinate-descent-attention
  Implementation of an Attention layer where each head can attend to more than just one token, using coordinate descent to pick topk
  ☆46Updated last year
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆45Updated last month
• expz / annotated-hyena
  An annotated implementation of the Hyena Hierarchy paper
  ☆30Updated last year
• facebookresearch / projUNN
  Fast training of unitary deep network layers from low-rank updates
  ☆28Updated last year
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆74Updated 7 months ago
• gregorbachmann / scaling_mlps
  ☆48Updated 3 months ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated last year
• epfml / schedules-and-scaling
  ☆47Updated 3 months ago
• GFNOrg / EB_GFN
  Code for our paper "Generative Flow Networks for Discrete Probabilistic Modeling"
  ☆73Updated last year
• lucidrains / jax2torch
  Use Jax functions in Pytorch
  ☆224Updated last year
• michaelsdr / sinkformers
  Transformers with doubly stochastic attention
  ☆40Updated 2 years ago
• Ping-C / optimizer
  This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…
  ☆34Updated last year
• Tikquuss / GflowNets_Tutorial
  GflowNets, MCMC, Metropolis-Hasting, Gibbs sampling, Metropolis-adjusted Langevin, Inverse Transform Sampling, Acceptance-Rejection Metho…
  ☆81Updated last year
• 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
• lucidrains / taylor-series-linear-attention
  Explorations into the recently proposed Taylor Series Linear Attention
  ☆85Updated last month
• Felix-Petersen / algovision
  Differentiable Algorithms and Algorithmic Supervision.
  ☆101Updated last year
• insuhan / hyper-attn
  ☆65Updated 9 months ago
• ag1988 / dss
  Sequence Modeling with Structured State Spaces
  ☆60Updated 2 years ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆62Updated last month
• EdinburghNLP / torch-adaptive-imle
  ☆32Updated 9 months ago
• joneswack / dp-rfs
  This repository contains PyTorch implementations of various random feature maps for dot product kernels.
  ☆17Updated 2 months ago
• irhum / hyena
  JAX/Flax implementation of the Hyena Hierarchy
  ☆29Updated last year
• radarFudan / mamba-minimal-jax
  ☆28Updated last week
• xu-ji / information-bottleneck
  Deep Learning & Information Bottleneck
  ☆45Updated last year
• tml-epfl / why-weight-decay
  Why Do We Need Weight Decay in Modern Deep Learning? [arXiv, Oct 2023]
  ☆41Updated 11 months ago