FLAIROx / jaxirlLinks
Contains JAX implementation of algorithms for inverse reinforcement learning
β72Updated last year
Alternatives and similar repositories for jaxirl
Users that are interested in jaxirl are comparing it to the libraries listed below
Sorting:
- β83Updated last week
- Baselines for gymnax π€β71Updated 2 years ago
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]β110Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weightsβ¦β58Updated 3 years ago
- Simple single-file baselines for Q-Learning in pure-GPU settingβ182Updated 6 months ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselinβ¦β56Updated this week
- A collection of RL algorithms written in JAX.β104Updated 3 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ114Updated last year
- β83Updated 10 months ago
- Unified Implementations of Offline Reinforcement Learning Algorithmsβ94Updated 4 months ago
- β46Updated 2 years ago
- JAX implementations of core Deep RL algorithmsβ82Updated 3 years ago
- Accelerated minigrid environments with JAXβ147Updated 2 weeks ago
- Accelerated replay buffers in JAXβ43Updated 3 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objectiveβ81Updated 2 years ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learningβ17Updated 2 years ago
- General Modules for JAXβ67Updated last week
- Corax: Core RL in JAXβ38Updated last year
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancyβ20Updated 10 months ago
- Challenging Memory-based Deep Reinforcement Learning Agentsβ103Updated 10 months ago
- An Open-Ended Agentic Simulatorβ52Updated last year
- JAX implementation of RL algorithms and vectorized environmentsβ48Updated last year
- Vectorization techniques for fast population-based training.β56Updated 3 years ago
- β41Updated last year
- β48Updated 2 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!β234Updated 3 months ago
- Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.β133Updated last year
- Evaluating long-term memory of reinforcement learning algorithmsβ148Updated 2 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)β12Updated 2 years ago
- On the model-based stochastic value gradient for continuous reinforcement learningβ56Updated 2 years ago