Matt00n / PolicyGradientsJax
On-Policy Policy Gradient Algorithms in JAX
☆20Updated 7 months ago
Related projects: ⓘ
- When Do Transformers Shine in RL? Decoupling Memory from Credit Assignment, NeurIPS 2023 (oral)☆49Updated 8 months ago
- Synthetic Experience Replay☆62Updated 3 months ago
- Extreme Q-Learning: Max Entropy RL without Entropy☆78Updated last year
- Learning diverse options through the Laplacian representation.☆22Updated 8 months ago
- Generalized Decision Transformer for Offline Hindsight Information Matching (ICLR2022)☆64Updated 2 years ago
- Learning to Modulate pre-trained Models in RL (Decision Transformer, LoRA, Fine-tuning)☆49Updated 8 months ago
- Repo for Implicit Diffusion Q-Learning☆85Updated 9 months ago
- Skeleton for scalable and flexible Jax RL implementations☆58Updated last year
- Foundation Policies with Hilbert Representations (ICML 2024)☆65Updated 5 months ago
- [NeurIPS 2023] Implementation of Elastic Decision Transformer☆28Updated 11 months ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆87Updated 3 months ago
- Code for TRANSDREAMER: REINFORCEMENT LEARNING WITH TRANSFORMER WORLD MODELS☆20Updated 11 months ago
- ☆25Updated last week
- Code for "Unsupervised Zero-Shot RL via Functional Reward Representations"☆51Updated 5 months ago
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆31Updated 6 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆71Updated 9 months ago
- Simple maze environments using mujoco-py☆52Updated 8 months ago
- Codebase for "Uni[MASK]: Unified Inference in Sequential Decision Problems"☆54Updated 2 months ago
- Transformer-based World Models☆66Updated last year
- [ICLR 2024] Closing the Gap between TD Learning and Supervised Learning - A Generalisation Point of View.☆19Updated 5 months ago
- Official codebase for GTA: Generative Trajectory Augmentation with Guidance for Offline Reinforcement Learning.☆11Updated last month
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆49Updated 11 months ago
- ☆27Updated last year
- Official code repository for Prompt-DT.☆93Updated 2 years ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆53Updated 3 months ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆100Updated 2 years ago
- Github repo for HIDIO: Hierarchical Reinforcement Learning by Discovering Intrinsic Options☆42Updated 2 years ago
- ☆37Updated last year
- Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]☆82Updated 9 months ago
- LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization☆31Updated last year