FLAIROx / jaxirlLinks
Contains JAX implementation of algorithms for inverse reinforcement learning
β73Updated last year
Alternatives and similar repositories for jaxirl
Users that are interested in jaxirl are comparing it to the libraries listed below
Sorting:
- β82Updated 5 months ago
- Baselines for gymnax π€β71Updated 2 years ago
- Simple single-file baselines for Q-Learning in pure-GPU settingβ181Updated 5 months ago
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselinβ¦β56Updated last week
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weightsβ¦β58Updated 3 years ago
- Accelerated replay buffers in JAXβ43Updated 2 years ago
- β81Updated 9 months ago
- Unified Implementations of Offline Reinforcement Learning Algorithmsβ92Updated 4 months ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRLβ114Updated last year
- JAX implementation of RL algorithms and vectorized environmentsβ47Updated 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
- β39Updated last year
- JAX implementations of core Deep RL algorithmsβ82Updated 3 years ago
- Challenging Memory-based Deep Reinforcement Learning Agentsβ103Updated 10 months ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancyβ18Updated 10 months ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learningβ17Updated 2 years ago
- Fast reinforcement learning researchβ61Updated 8 months ago
- General Modules for JAXβ67Updated 4 months ago
- A collection of RL algorithms written in JAX.β103Updated 3 years ago
- An Open-Ended Agentic Simulatorβ52Updated last year
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"β64Updated 3 months ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorchβ52Updated 2 years ago
- Accelerated minigrid environments with JAXβ144Updated 2 weeks ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!β232Updated 3 months ago
- β46Updated 2 years ago
- Code for Discovered Policy Optimisation (NeurIPS 2022)β11Updated 2 years ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objectiveβ80Updated 2 years ago
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"β58Updated last year
- Corax: Core RL in JAXβ38Updated last year
- A project that provides help for using DeepMind's mctx on gym-style environments.β60Updated 9 months ago