seijin-kobayashi / cocoa
Code accompanying the paper "Would I have gotten that reward? Long-term credit assignment by counterfactual contribution analysis"
☆11Updated last year
Alternatives and similar repositories for cocoa:
Users that are interested in cocoa are comparing it to the libraries listed below
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Updated 2 years ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆19Updated 3 years ago
- Code that can be used to reproduce the experiments in our paper "Estimating Risk and Uncertainty in Deep Reinforcement Learning"☆30Updated 2 years ago
- An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.☆15Updated 2 years ago
- Official repo for Offline RL for Online RL☆16Updated last year
- [AutoML'22] Bayesian Generational Population-based Training (BG-PBT)☆28Updated 2 years ago
- ☆15Updated last year
- ☆16Updated 2 years ago
- Official PyTorch Implementation for Metric Residual Networks for Sample Efficient Goal-Conditioned Reinforcement Learning☆17Updated 2 years ago
- POPGym Library in JAX☆11Updated last year
- ☆16Updated 3 years ago
- Official implementation of Harnessing Mixed Offline Reinforcement Learning Datasets via Trajectory Reweighting☆16Updated last year
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆16Updated 10 months ago
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆15Updated 2 years ago
- Implementation of Proximal Policy Optimization in Jax+Flax☆18Updated last year
- ☆31Updated 4 years ago
- ☆24Updated 8 months ago
- SkillHack: A Benchmark for Skill Transfer in Open-Ended Reinforcement Learning☆14Updated 2 years ago
- Code for the paper "Uncertainty-Driven Exploration for Generalization in Reinforcement Learning".☆27Updated last year
- Code for paper "Bridging Imagination and Reality for Model-Based Deep Reinforcement Learning".☆14Updated 3 years ago
- Implementation for ICML 2019 paper, EMI: Exploration with Mutual Information.☆36Updated 4 years ago
- ☆32Updated 8 months ago
- IV-RL - Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation☆40Updated 5 months ago
- ☆21Updated last year
- Author's PyTorch implementation of SR-DICE for marginalized importance sampling☆16Updated 3 years ago
- Change-Based Exploration Transfer☆36Updated 2 years ago
- Toolkit of Causal Model-based Reinforcement Learning.☆33Updated last year
- Scalable Opponent Shaping Experiments in JAX☆24Updated last year
- Official code repo for paper: Hybrid RL: Using both offline and online data can make RL efficient.☆24Updated 2 years ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆29Updated last year