fabian-sp / MoMoLinks
MoMo: Momentum Models for Adaptive Learning Rates
☆19Updated last year
Alternatives and similar repositories for MoMo
Users that are interested in MoMo are comparing it to the libraries listed below
Sorting:
- ☆120Updated 3 months ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆57Updated last month
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆79Updated last year
- Comparison between GFlowNets & Maximum Entropy RL☆19Updated last year
- A dataloader, but for JAX☆20Updated last year
- Pytorch-like dataloaders for JAX.☆94Updated 4 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆102Updated last month
- ☆302Updated 8 months ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆110Updated last year
- ☆33Updated 10 months ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆31Updated 6 months ago
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆71Updated 5 months ago
- Dreamer on JAX☆16Updated 3 years ago
- Optimal Transport for Machine Learners☆30Updated 4 months ago
- Flow-matching algorithms in JAX☆105Updated last year
- Parallelizing non-linear sequential models over the sequence length☆54Updated 3 months ago
- ☆25Updated 9 months ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆81Updated last year
- Scalable and Stable Parallelization of Nonlinear RNNS☆22Updated last month
- Unified Implementations of Offline Reinforcement Learning Algorithms☆111Updated 5 months ago
- [ICLR 2025] Implementation of "FACTS: A Factored State-Space Framework For World Modelling"☆28Updated 4 months ago
- JAX implementation of VQVAE/VQGAN autoencoders (+FSQ)☆37Updated last year
- Implementation of PSGD optimizer in JAX☆33Updated 9 months ago
- A simple library for scaling up JAX programs☆143Updated 11 months ago
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆99Updated last month
- ☆84Updated 11 months ago
- ☆71Updated last year
- Oscillatory State-Space Models☆100Updated 6 months ago
- Agar.io for Continual Reinforcement Learning☆22Updated 2 months ago