Alternatives and similar repositories for MoMo

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

• NicolasZucchet / minimal-LRU
  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)
  ☆54Updated 7 months ago
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆18Updated last year
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆101Updated last year
• Artur-Galstyan / jaxonloader
  A dataloader, but for JAX
  ☆21Updated last year
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆83Updated last week
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 5 months ago
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆11Updated last year
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 5 months ago
• kvfrans / jax-flow
  Flow-matching algorithms in JAX
  ☆92Updated 9 months ago
• lockwo / distreqx
  Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.
  ☆31Updated last month
• ruke1ire / RTF
  A State-Space Model with Rational Transfer Function Representation.
  ☆78Updated last year
• i404788 / s5-pytorch
  Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)
  ☆74Updated last year
• machine-discovery / deer
  Parallelizing non-linear sequential models over the sequence length
  ☆51Updated 4 months ago
• xl0 / lovely-jax
  JAX Arrays for human consumption
  ☆91Updated last year
• samuela / torch2jax
  Run PyTorch in JAX. 🤝
  ☆246Updated 3 months ago
• Benjamin-Walker / selective-ssms-and-linear-cdes
  Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)
  ☆12Updated 4 months ago
• google-deepmind / spectral_ssm
  ☆32Updated last year
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆35Updated 3 weeks ago
• proger / accelerated-scan
  Accelerated First Order Parallel Associative Scan
  ☆181Updated 9 months ago
• 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…
  ☆26Updated 6 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆137Updated 7 months ago
• alisiahkoohi / Langevin-dynamics
  Sampling with gradient-based Markov Chain Monte Carlo approaches
  ☆101Updated last year
• modula-systems / modula
  🧱 Modula software package
  ☆194Updated 2 months ago
• NX-AI / flashrnn
  FlashRNN - Fast RNN Kernels with I/O Awareness
  ☆90Updated 2 months ago
• autonomousvision / akorn
  [ICLR'25] Artificial Kuramoto Oscillatory Neurons
  ☆89Updated 2 weeks ago
• twitter-research / hyperbolic-rl
  ☆56Updated 2 years ago
• facebookresearch / SOC-matching
  Code release for "Stochastic Optimal Control Matching"
  ☆35Updated 9 months ago
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆33Updated last year
• johnryan465 / pscan
  ☆40Updated last year
• yardenas / jax-dreamer
  Dreamer on JAX
  ☆16Updated 3 years ago