Alternatives and similar repositories for MoMo

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

• kvfrans / splus
  ☆122Updated 6 months ago
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆19Updated last year
• smorad / memax
  Deep memory and sequence models in JAX
  ☆20Updated 2 weeks ago
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆20Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆309Updated 2 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆112Updated last year
• NicolasZucchet / minimal-LRU
  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)
  ☆61Updated 3 months ago
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆111Updated 2 years ago
• gpeyre / ot4ml
  Optimal Transport for Machine Learners
  ☆42Updated last month
• acerbilab / amortized-conditioning-engine
  Amortized Probabilistic Conditioning for Optimization, Simulation and Inference (Chang et al., AISTATS 2025)
  ☆21Updated 6 months ago
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆97Updated last week
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated last month
• keraJLi / rejax
  Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!
  ☆251Updated last month
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆35Updated 11 months ago
• Gothos / LRU-pytorch
  Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch
  ☆72Updated 8 months ago
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• sukhijab / maxinforl_jax
  ☆27Updated 2 months ago
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆37Updated 2 years ago
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆41Updated 7 months ago
• EmptyJackson / unifloral
  Unified Implementations of Offline Reinforcement Learning Algorithms
  ☆182Updated last week
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• NanboLi / FACTS
  [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"
  ☆28Updated 6 months ago
• dunnolab / xland-minigrid-datasets
  XLand-100B: A Large-Scale Multi-Task Dataset for In-Context Reinforcement Learning - - — ICLR 2025
  ☆81Updated 10 months ago
• nmboffi / jax-interpolants
  Definitive implementation of the stochastic interpolant framework for generative modeling in jax.
  ☆39Updated 5 months ago
• willisma / diffuse_nnx
  A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…
  ☆125Updated 2 months ago
• lindermanlab / S5
  ☆312Updated 11 months ago
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆153Updated 2 months ago
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Updated last year
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆56Updated 6 months ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆106Updated 2 months ago