subho406 / Recurrent-PPO-JaxView external linksLinks
Implementation of Proximal Policy Optimization in Jax+Flax
☆21May 18, 2023Updated 2 years ago
Alternatives and similar repositories for Recurrent-PPO-Jax
Users that are interested in Recurrent-PPO-Jax are comparing it to the libraries listed below
Sorting:
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafter☆18Nov 29, 2025Updated 2 months ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated last year
- ☆10Jun 27, 2024Updated last year
- Official Implementation of `An Optimisation Framework for Unsupervised Environment Design` from RLC 2025☆17Nov 24, 2025Updated 2 months ago
- ☆13Apr 25, 2024Updated last year
- Reproduction of Curiosity-driven Exploration by Self-supervised Prediction in PyTorch☆13Jun 10, 2019Updated 6 years ago
- Gym env for Slay the Spire☆16Dec 31, 2024Updated last year
- Opinionated library for managing hyperparameters and mutable state of machine learning training systems.☆19Aug 4, 2023Updated 2 years ago
- Decision Transformer JAX - Reproduction of 'Decision Transformer: Reinforcement Learning via Sequence Modeling' in JAX and Haiku☆12Aug 14, 2024Updated last year
- ☆16Jul 16, 2024Updated last year
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆30Oct 29, 2023Updated 2 years ago
- ☆15Jul 16, 2021Updated 4 years ago
- Model-Based RL Demo for Pendulum-v0☆13Jun 16, 2020Updated 5 years ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆15Jan 3, 2023Updated 3 years ago
- A collection of matrix games in JAX☆13Nov 28, 2024Updated last year
- Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.☆21Feb 20, 2023Updated 2 years ago
- ☆15Sep 28, 2022Updated 3 years ago
- Adding Dreamer-v3's implementation tricks to CleanRL's PPO☆14May 19, 2023Updated 2 years ago
- ☆18Apr 22, 2024Updated last year
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆12Aug 31, 2020Updated 5 years ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆20Mar 22, 2022Updated 3 years ago
- ☆35Nov 22, 2024Updated last year
- Baseline implementation of recurrent PPO using truncated BPTT☆160Apr 28, 2024Updated last year
- JAX implementation of RL algorithms and vectorized environments☆51Dec 26, 2023Updated 2 years ago
- Continual reinforcement learning baselines: experiment specifications, implementation of existing methods, and common metrics. Easily ext…☆131Jul 6, 2023Updated 2 years ago
- ☆13Jan 16, 2025Updated last year
- Code to accompany the paper "Mismatched No More: Joint Model-Policy Optimization for Model-Based RL"☆20Oct 6, 2021Updated 4 years ago
- AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)☆23Oct 15, 2024Updated last year
- ☆46Sep 24, 2024Updated last year
- A benchmark to test and compare your pcg algorithm against each other☆22Sep 22, 2025Updated 4 months ago
- ☆19Mar 1, 2023Updated 2 years ago
- Official pytorch implementation for our ICLR 2023 paper "Latent State Marginalization as a Low-cost Approach for Improving Exploration".☆24Feb 9, 2023Updated 3 years ago
- A number of agents (PPO, MuZero) with a Perceiver-based NN architecture that can be trained to achieve goals in nethack/minihack environm…☆43Sep 19, 2022Updated 3 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆23Oct 28, 2024Updated last year
- ☆20May 22, 2022Updated 3 years ago
- RL-Toolkit: A Research Framework for Robotics☆21Jan 22, 2026Updated 3 weeks ago
- Fast dataset format and loader☆23Jan 2, 2026Updated last month
- A minimal home grid world environment to evaluate language understanding in interactive agents.☆24Sep 6, 2023Updated 2 years ago