fabian-sp / MoMoLinks
MoMo: Momentum Models for Adaptive Learning Rates
☆19Updated 11 months ago
Alternatives and similar repositories for MoMo
Users that are interested in MoMo are comparing it to the libraries listed below
Sorting:
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆54Updated 7 months ago
- Comparison between GFlowNets & Maximum Entropy RL☆18Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆101Updated last year
- A dataloader, but for JAX☆21Updated last year
- Pytorch-like dataloaders for JAX.☆83Updated last week
- Lightning-like training API for JAX with Flax☆38Updated 5 months ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆11Updated last year
- Implementation of PSGD optimizer in JAX☆33Updated 5 months ago
- Flow-matching algorithms in JAX☆92Updated 9 months ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆31Updated last month
- A State-Space Model with Rational Transfer Function Representation.☆78Updated last year
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆74Updated last year
- Parallelizing non-linear sequential models over the sequence length☆51Updated 4 months ago
- JAX Arrays for human consumption☆91Updated last year
- Run PyTorch in JAX. 🤝☆246Updated 3 months ago
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆12Updated 4 months ago
- ☆32Updated last year
- Efficiently Composable Data Augmentation on the GPU with Jax☆35Updated 3 weeks ago
- Accelerated First Order Parallel Associative Scan☆181Updated 9 months ago
- Codebase to fully reproduce the results of "No Representation, No Trust: Connecting Representation, Collapse, and Trust Issues in PPO" (M…☆26Updated 6 months ago
- A simple library for scaling up JAX programs☆137Updated 7 months ago
- Sampling with gradient-based Markov Chain Monte Carlo approaches☆101Updated last year
- 🧱 Modula software package☆194Updated 2 months ago
- FlashRNN - Fast RNN Kernels with I/O Awareness☆90Updated 2 months ago
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆89Updated 2 weeks ago
- ☆56Updated 2 years ago
- Code release for "Stochastic Optimal Control Matching"☆35Updated 9 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆33Updated last year
- ☆40Updated last year
- Dreamer on JAX☆16Updated 3 years ago