m-wojnar / reinforced-lib
Reinforcement learning library
☆64Updated last week
Related projects ⓘ
Alternatives and complementary repositories for reinforced-lib
- Jaxpr Visualisation Tool☆16Updated 3 months ago
- ☆48Updated last year
- Classic MCTS example with mctx☆16Updated last year
- Giving Up Control: Neurons as Reinforcement Learning Agents☆11Updated 6 months ago
- A collection of meta-learning algorithms in Jax☆23Updated 2 years ago
- Easy Hypernetworks in Pytorch and Jax☆96Updated last year
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆50Updated last year
- MoMo: Momentum Models for Adaptive Learning Rates☆17Updated 5 months ago
- Latent Program Network (from the "Searching Latent Program Spaces" paper)☆28Updated last week
- Accelerated replay buffers in JAX☆40Updated 2 years ago
- Implementation (TensorFlow/keras) of the DreamerV3 model-based RL algorithm by Hafner et al. 2023☆3Updated last year
- fast + parallel AlphaZero in JAX☆85Updated 7 months ago
- Gradient-based constrained optimization for JAX☆26Updated 2 years ago
- A dataloader, but for JAX☆18Updated 6 months ago
- Code for the papers Hypernetworks in Meta-Reinforcement Learning (Beck et al., 2022) and Recurrent Hypernetworks are Surprisingly Strong …☆10Updated 3 months ago
- Code for magnetic mirror descent.☆15Updated last year
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆46Updated last year
- Accelerated minigrid environments with JAX☆120Updated 3 months ago
- ADAHESSIAN: An Adaptive Second Order Optimizer for Machine Learning☆266Updated last year
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆12Updated 4 months ago
- A collection of matrix games in JAX☆10Updated 2 weeks ago
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆42Updated 4 months ago
- Educational material for teaching the random bag of tricks needed to program real-time systems.☆12Updated 2 months ago
- Implementing RASP transformer programming language https://arxiv.org/pdf/2106.06981.pdf.☆44Updated 3 years ago
- ☆105Updated 2 weeks ago
- Develop your agent for generals.io!☆35Updated this week
- A project that provides help for using DeepMind's mctx on gym-style environments.☆50Updated last week
- Train to 94% on CIFAR-10 in 4.4 seconds on a single A100☆12Updated 10 months ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆36Updated 2 months ago
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆48Updated 2 weeks ago