ShaneFlandermeyer / tdmpc2-jax
Jax/Flax Implementation of TD-MPC2
☆46Updated 2 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for tdmpc2-jax
- OGBench: Benchmarking Offline Goal-Conditioned RL☆72Updated last week
- ☆24Updated 3 months ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆72Updated 5 months ago
- (ICLR 2024) Reverse Forward Curriculum Learning☆37Updated 2 months ago
- Code for SAPG: Split and Aggregate Policy Gradients (ICML 2024)☆41Updated last month
- Parallel Q-Learning: Scaling Off-policy Reinforcement Learning under Massively Parallel Simulation☆61Updated last year
- A minimal and stable PPO.☆117Updated 9 months ago
- PWM: Policy Learning with Large World Models☆37Updated 2 months ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆57Updated 5 months ago
- ☆42Updated last year
- ☆79Updated this week
- Demo-Driven Mobile Bi-Manual Manipulation Benchmark.☆96Updated 3 weeks ago
- Source files to replicate experiments in my ICLR 2022 paper.☆61Updated 4 months ago
- Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)☆16Updated 5 months ago
- Code and website for Behavior Transformers: Cloning k modes with one stone.☆109Updated last year
- Skill-based Model-based Reinforcement Learning (CoRL 2022)☆52Updated 2 years ago
- Official code for "Behavior Generation with Latent Actions" (ICML 2024 Spotlight)☆109Updated 8 months ago
- ☆52Updated last year
- [ICLR 2024] Official implementation for "Towards Diverse Behaviors: A Benchmark for Imitation Learning with Human Demonstrations"☆43Updated 5 months ago
- Collection of MuJoCo robotics environments equipped with both vision and tactile sensing☆40Updated 4 months ago
- DrM, a visual RL algorithm, minimizes the dormant ratio to guide exploration-exploitation trade-offs, achieving significant improvements …☆61Updated 5 months ago
- Skeleton for scalable and flexible Jax RL implementations☆63Updated last year
- Wrappers and utilities for Nvidia IsaacGym☆92Updated 2 years ago
- Efficient Real-World RL for Legged Locomotion via Adaptive Policy Regularization☆63Updated last year
- Code release for Efficient Planning in a Compact Latent Action Space (ICLR2023) https://arxiv.org/abs/2208.10291.☆94Updated last year
- METRA: Scalable Unsupervised RL with Metric-Aware Abstraction (ICLR 2024)☆57Updated last year
- ☆69Updated 2 years ago
- ☆53Updated 3 weeks ago
- Finetuning Offline World Models in the Real World☆47Updated last year
- Coarse-to-fine Q-Network☆29Updated 3 months ago