Alternatives and similar repositories for MoMo

Users that are interested in MoMo are comparing it to the libraries listed below

• kvfrans / splus
  ☆104Updated 2 weeks ago
• NicolasZucchet / minimal-LRU
  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)
  ☆53Updated 7 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆41Updated 6 months ago
• lebrice / torch_jax_interop
  Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!
  ☆31Updated last month
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆103Updated last year
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆18Updated last year
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆139Updated 7 months ago
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆21Updated last year
• CLAIRE-Labo / no-representation-no-trust
  Codebase to fully reproduce the results of "No Representation, No Trust: Connecting Representation, Collapse, and Trust Issues in PPO" (M…
  ☆27Updated 7 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆87Updated 3 weeks ago
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆252Updated 4 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆97Updated 10 months ago
• radarFudan / mamba-minimal-jax
  ☆31Updated 7 months ago
• yardenas / jax-dreamer
  Dreamer on JAX
  ☆16Updated 3 years ago
• danijar / elements
  Building blocks for productive research
  ☆58Updated 4 months ago
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆80Updated 2 months ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆31Updated 2 months ago
• Gothos / LRU-pytorch
  Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch
  ☆70Updated 2 months ago
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆11Updated last year
• kvfrans / jax-vqvae-vqgan
  JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)
  ☆30Updated last year
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆93Updated last week
• alisiahkoohi / Langevin-dynamics
  Sampling with gradient-based Markov Chain Monte Carlo approaches
  ☆103Updated last year
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆91Updated last week
• Reytuag / transformerXL_PPO_JAX
  ☆80Updated 7 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 5 months ago
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆37Updated last month
• danijar / ninjax
  General Modules for JAX
  ☆65Updated 2 months ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• sukhijab / maxinforl_jax
  ☆23Updated 6 months ago
• dojeon-ai / SimbaV2
  Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"
  ☆51Updated 3 weeks ago