zhouzypaul / wsrl
JAX implementation of WSRL and RL baselines | ICLR 2025
☆22Updated last month
Alternatives and similar repositories for wsrl:
Users that are interested in wsrl are comparing it to the libraries listed below
- Official codebase for "Privileged Sensing Scaffolds Reinforcement Learning", contains the Scaffolder algorithm and Sensory Scaffolding Su…☆25Updated 9 months ago
- DrM, a visual RL algorithm, minimizes the dormant ratio to guide exploration-exploitation trade-offs, achieving significant improvements …☆69Updated 8 months ago
- ☆24Updated last year
- Official code for TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representations☆26Updated 4 months ago
- PWM: Policy Learning with Large World Models☆41Updated 6 months ago
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆76Updated 2 months ago
- ☆35Updated 3 weeks ago
- Code for SAPG: Split and Aggregate Policy Gradients (ICML 2024)☆46Updated 5 months ago
- [ICML 2023] Pre-train world model-based agents with different unsupervised strategies, fine-tune the agent's components selectively, and …☆37Updated 11 months ago
- [NeurIPS 2024] GenRL: Multimodal-foundation world models enable grounding language and video prompts into embodied domains, by turning th…☆68Updated last month
- Jax/Flax Implementation of TD-MPC2☆55Updated this week
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆62Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆64Updated 8 months ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆98Updated 8 months ago
- A benchmark for offline goal-conditioned RL and offline RL☆120Updated last week
- ☆20Updated 9 months ago
- Parallel Q-Learning: Scaling Off-policy Reinforcement Learning under Massively Parallel Simulation☆66Updated last year
- Implementation of Jump-Start Reinforcement Learning (JSRL) with Stable Baselines3☆29Updated last year
- (ICLR 2024) Reverse Forward Curriculum Learning☆42Updated 2 months ago
- ☆72Updated 8 months ago
- Skill-based Model-based Reinforcement Learning (CoRL 2022)☆55Updated 2 years ago
- From Play to Policy: Conditional Behavior Generation from Uncurated Robot Data☆53Updated last year
- Code for "Masked Autoencoding for Scalable and Generalizable Decision Making". NeurIPS 2022☆43Updated 11 months ago
- Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)☆17Updated 8 months ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆78Updated 9 months ago
- ☆24Updated 7 months ago
- Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024☆62Updated 8 months ago
- Controllability-Aware Unsupervised Skill Discovery (ICML 2023)☆23Updated last year
- Foundation Policies with Hilbert Representations (ICML 2024)☆76Updated 10 months ago
- ☆70Updated 4 months ago