Matt00n / PolicyGradientsJaxLinks
On-Policy Policy Gradient Algorithms in JAX
☆41Updated last year
Alternatives and similar repositories for PolicyGradientsJax
Users that are interested in PolicyGradientsJax are comparing it to the libraries listed below
Sorting:
- Extreme Q-Learning: Max Entropy RL without Entropy☆87Updated 2 years ago
- A general model-free off-policy actor-critic implementation. Continuous and Discrete Soft Actor-Critic with multimodal observations, data…☆40Updated last year
- Benchmarked implementations of Offline RL Algorithms.☆76Updated 9 months ago
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆68Updated last year
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆61Updated last year
- Foundation Policies with Hilbert Representations (ICML 2024)☆103Updated 2 months ago
- Synthetic Experience Replay☆107Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆81Updated last year
- Repo for Implicit Diffusion Q-Learning☆119Updated 2 years ago
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆79Updated last month
- Source files to replicate experiments in my ICLR 2022 paper.☆70Updated 5 months ago
- Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.☆110Updated 2 years ago
- Author's Pytorch implementation of ICLR2023 paper Behavior Proximal Policy Optimization (BPPO).☆93Updated 2 years ago
- Author's PyTorch implementation of TD7 for online and offline RL☆157Updated 2 years ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆69Updated 3 years ago
- ☆113Updated 9 months ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆103Updated 3 years ago
- Transformer-based World Models☆86Updated 2 years ago
- ☆14Updated last year
- ☆47Updated last month
- [NeurIPS 2024] Maximum Entropy Reinforcement Learning via Energy-Based Normalizing Flow☆40Updated last year
- An unofficial implementation for online decision transformer☆40Updated 3 years ago
- ☆58Updated last week
- Challenging Memory-based Deep Reinforcement Learning Agents☆105Updated last year
- A pytorch reprelication of the model-based reinforcement learning algorithm MBPO☆182Updated 3 years ago
- MR.Q is a general-purpose model-free reinforcement learning algorithm.☆129Updated 5 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆92Updated last year
- Implementation of Trajectory Transformer with attention caching and batched beam search☆116Updated 2 years ago
- official implementation for our paper Cal-QL: Calibrated Offline RL Pre-Training for Efficient Online Fine-Tuning (NeurIPS 2023)☆115Updated last year
- ☆300Updated 3 years ago