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 4 months ago
- A dataloader, but for JAX☆20Updated last year
- Pytorch-like dataloaders for JAX.☆93Updated 5 months ago
- Flow-matching algorithms in JAX☆106Updated last year
- ☆34Updated 11 months ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆58Updated last month
- Implementation of PSGD optimizer in JAX☆35Updated 9 months ago
- A simple library for scaling up JAX programs☆144Updated 11 months ago
- Run PyTorch in JAX. 🤝☆305Updated 2 weeks ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Updated last year
- Deep memory and sequence models in JAX☆15Updated last week
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆31Updated 7 months ago
- Comparison between GFlowNets & Maximum Entropy RL☆19Updated last year
- [ICLR'25] Artificial Kuramoto Oscillatory Neurons☆104Updated last week
- Implementation of the new SOTA for model based RL, from the paper "Improving Transformer World Models for Data-Efficient RL", in Pytorch☆141Updated 5 months ago
- Lightning-like training API for JAX with Flax☆44Updated 10 months ago
- ☆24Updated 10 months ago
- Optimal Transport for Machine Learners☆33Updated 5 months ago
- Pytorch implementation of Simplified Structured State-Spaces for Sequence Modeling (S5)☆79Updated last year
- Pytorch implementation of Evolutionary Policy Optimization, from Wang et al. of the Robotics Institute at Carnegie Mellon University☆103Updated last month
- ☆306Updated 9 months ago
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆106Updated 2 weeks ago
- Minimal, lightweight JAX implementations of popular models.☆114Updated last week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆37Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆110Updated last year
- A State-Space Model with Rational Transfer Function Representation.☆82Updated last year
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆71Updated 6 months ago
- Minimal yet performant LLM examples in pure JAX☆187Updated last month
- Efficiently Composable Data Augmentation on the GPU with Jax☆41Updated 5 months ago
- Definitive implementation of the stochastic interpolant framework for generative modeling in jax.☆32Updated 3 months ago