Matt00n / PolicyGradientsJaxLinks
On-Policy Policy Gradient Algorithms in JAX
☆37Updated last year
Alternatives and similar repositories for PolicyGradientsJax
Users that are interested in PolicyGradientsJax are comparing it to the libraries listed below
Sorting:
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆62Updated last year
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆58Updated 8 months ago
- Extreme Q-Learning: Max Entropy RL without Entropy☆88Updated 2 years ago
- Benchmarked implementations of Offline RL Algorithms.☆73Updated 3 months ago
- Official codebase for GTA: Generative Trajectory Augmentation with Guidance for Offline Reinforcement Learning.☆22Updated 7 months ago
- ☆93Updated 4 months ago
- Source files to replicate experiments in my ICLR 2022 paper.☆70Updated 11 months ago
- Author's Pytorch implementation of ICLR2023 paper Behavior Proximal Policy Optimization (BPPO).☆87Updated last year
- Synthetic Experience Replay☆92Updated last year
- An unofficial implementation for online decision transformer☆40Updated 2 years ago
- ☆104Updated 2 years ago
- Repo for Implicit Diffusion Q-Learning☆109Updated last year
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆103Updated last year
- Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.☆108Updated 2 years ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆67Updated 2 years ago
- Code for NeurIPS2023 accepted paper: Counterfactual Conservative Q Learning for Offline Multi-agent Reinforcement Learning.☆36Updated 4 months ago
- [NeurIPS 2024] Maximum Entropy Reinforcement Learning via Energy-Based Normalizing Flow☆36Updated 7 months ago
- Code for "SimbaV2: Hyperspherical Normalization for Scalable Deep Reinforcement Learning"☆51Updated 3 weeks ago
- Transformer-based World Models☆82Updated 2 years ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆85Updated 6 months ago
- Scaling Pareto-Efficient Decision Making via Offline Multi-Objective RL, published in ICLR 2023☆32Updated 6 months ago
- PyTorch implementation of our paper Reinforcement Learning with Random Delays (ICLR 2020)☆41Updated 3 years ago
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆70Updated last year
- [ICML 2024] The algorithm of Reinforcement Learning with an Assistant Reward Agent (ReLara)☆13Updated 10 months ago
- Pre-Training Goal-based Models for Sample-Efficient Reinforcement Learning☆27Updated last year
- Unified Implementations of Offline Reinforcement Learning Algorithms☆80Updated 2 months ago
- 🔥 Datasets and env wrappers for offline safe reinforcement learning☆94Updated 9 months ago
- Repository for "Quality-Diversity Actor-Critic: Learning High-Performing and Diverse Behaviors via Value and Successor Features Critics" …☆16Updated last year
- Implementation of Multi-Game Decision Transformers in PyTorch☆47Updated 2 years ago
- Author's PyTorch implementation of TD7 for online and offline RL☆145Updated last year