Alternatives and similar repositories for Online-learning-LR-dependencies

Users that are interested in Online-learning-LR-dependencies are comparing it to the libraries listed below

• NicolasZucchet / minimal-LRU
  Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)
  ☆61Updated 4 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆35Updated last year
• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆24Updated 3 years ago
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆19Updated last year
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆112Updated 2 years ago
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆37Updated 2 years ago
• lebrice / torch_jax_interop
  Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!
  ☆35Updated 3 weeks ago
• meulemansalex / theoretical_framework_for_target_propagation
  Python implementation of the methods in Meulemans et al. 2020 - A Theoretical Framework For Target Propagation
  ☆31Updated last year
• johnryan465 / pscan
  ☆40Updated 2 years ago
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆46Updated 3 years ago
• google-research / jaxpruner
  ☆234Updated 11 months ago
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆57Updated 3 years ago
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆156Updated 3 months ago
• young-geng / mintext
  Minimal but scalable implementation of large language models in JAX
  ☆35Updated 2 months ago
• yardenas / jax-dreamer
  Dreamer on JAX
  ☆16Updated 4 years ago
• danijar / ninjax
  General Modules for JAX
  ☆72Updated 4 months ago
• jenkspt / gpt-jax
  Jax/Flax rewrite of Karpathy's nanoGPT
  ☆63Updated 2 years ago
• andyljones / boardlaw
  Scaling scaling laws with board games.
  ☆53Updated 2 years ago
• DramaCow / jaxued
  ☆91Updated last week
• luchris429 / JaxLife
  An Open-Ended Agentic Simulator
  ☆58Updated last year
• alecwangcq / KFAC-Pytorch
  Pytorch implementation of KFAC and E-KFAC (Natural Gradient).
  ☆133Updated 6 years ago
• TheodoreWolf / hyperoptax
  Parallel hyperparameter tuning with JAX
  ☆39Updated 6 months ago
• young-geng / scalax
  A simple library for scaling up JAX programs
  ☆144Updated 2 months ago
• AllanYangZhou / universal_neural_functional
  ☆55Updated last year
• henry-prior / jax-rl
  JAX implementations of core Deep RL algorithms
  ☆82Updated 3 years ago
• Reytuag / transformerXL_PPO_JAX
  ☆90Updated last year
• radarFudan / mamba-minimal-jax
  ☆35Updated last year
• vivekmyers / contrastive_planning
  Code for the paper "Inference via Interpolation: Contrastive Representations Provably Enable Planning and Inference"
  ☆43Updated last year
• lowrollr / mctx-az
  Monte Carlo tree search in JAX, with functionality to continue search from a previous subtree
  ☆26Updated 9 months ago
• world-modelz / world-modelz
  video prediction and world model research
  ☆14Updated 3 years ago