Matt00n / PolicyGradientsJax
On-Policy Policy Gradient Algorithms in JAX
☆30Updated last year
Alternatives and similar repositories for PolicyGradientsJax:
Users that are interested in PolicyGradientsJax are comparing it to the libraries listed below
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆62Updated last year
- Extreme Q-Learning: Max Entropy RL without Entropy☆85Updated 2 years ago
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆55Updated 5 months ago
- Benchmarked implementations of Offline RL Algorithms.☆72Updated 3 weeks ago
- Meta-RL Model-Based Algorithm☆31Updated 10 months ago
- Official codebase for GTA: Generative Trajectory Augmentation with Guidance for Offline Reinforcement Learning.☆19Updated 4 months ago
- Synthetic Experience Replay☆89Updated 10 months ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆67Updated 2 years ago
- Corax: Core RL in JAX☆37Updated last year
- ☆41Updated last year
- Source files to replicate experiments in my ICLR 2022 paper.☆70Updated 8 months ago
- Transformer-based World Models☆78Updated last year
- IV-RL - Sample Efficient Deep Reinforcement Learning via Uncertainty Estimation☆40Updated 5 months ago
- PyTorch implementation of our paper Reinforcement Learning with Random Delays (ICLR 2020)☆40Updated 2 years ago
- Scaling Pareto-Efficient Decision Making via Offline Multi-Objective RL, published in ICLR 2023☆32Updated 3 months ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆25Updated last year
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆40Updated last week
- Jax-Baseline is a Reinforcement Learning implementation using JAX and Flax/Haiku libraries, mirroring the functionality of Stable-Baselin…☆48Updated last week
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆15Updated 9 months ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆68Updated last year
- Foundation Policies with Hilbert Representations (ICML 2024)☆80Updated 11 months ago
- Author's Pytorch implementation of ICLR2023 paper Behavior Proximal Policy Optimization (BPPO).☆82Updated last year
- Repo for Implicit Diffusion Q-Learning☆104Updated last year
- PyTorch Implementation of the Maximum a Posteriori Policy Optimisation☆74Updated 2 years ago
- Pytorch implementation of "Maximum a Posteriori Policy Optimization" with Retrace for Discrete gym environments☆27Updated 4 years ago
- Code for NeurIPS2023 accepted paper: Counterfactual Conservative Q Learning for Offline Multi-agent Reinforcement Learning.☆34Updated last month
- Official repository for paper "Versatile Offline Imitation from Observations and Examples via Regularized State-Occupancy Matching" (ICML…☆25Updated 2 years ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆78Updated 3 months ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆67Updated 9 months ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆111Updated 3 years ago