kinalmehta / marl-jax
JAX library for MARL research
☆82Updated 10 months ago
Related projects: ⓘ
- Evaluating long-term memory of reinforcement learning algorithms☆129Updated last year
- PAIRED in PyTorch 🔥☆56Updated last year
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆96Updated 2 years ago
- ☆56Updated 3 weeks ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆102Updated 3 weeks ago
- Deep Hierarchical Planning from Pixels☆85Updated last year
- Implementation of Trajectory Transformer with attention caching and batched beam search☆101Updated last year
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆76Updated last year
- Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"☆41Updated 2 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆87Updated last month
- Benchmarking RL generalization in an interpretable way.☆128Updated 7 months ago
- A collection of RL algorithms written in JAX.☆92Updated 2 years ago
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.☆121Updated last month
- Revisiting Rainbow☆73Updated 3 years ago
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆49Updated 2 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆59Updated last month
- ☆100Updated 7 months ago
- Challenging Memory-based Deep Reinforcement Learning Agents☆76Updated 4 months ago
- Baselines for gymnax 🤖☆57Updated last year
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆46Updated last year
- ☆41Updated 5 months ago
- Open source code for paper "Denoised MDPs: Learning World Models Better Than the World Itself"☆133Updated last year
- impact-driven-exploration☆125Updated 11 months ago
- Partially Observable Process Gym☆158Updated 2 months ago
- A categorised list of Multi-Agent Reinforcemnt Learning (MARL) papers☆46Updated last year
- official implementation for our paper Cal-QL: Calibrated Offline RL Pre-Training for Efficient Online Fine-Tuning☆68Updated last month
- This code implements Prioritized Level Replay, a method for sampling training levels for reinforcement learning agents that exploits the …☆83Updated 3 years ago
- [NeurIPS 2022] Open source code for reusing prior computational work in RL.☆91Updated last year
- ☆25Updated last week