ShaneFlandermeyer / tdmpc2-jax
Jax/Flax Implementation of TD-MPC2
☆58Updated this week
Alternatives and similar repositories for tdmpc2-jax:
Users that are interested in tdmpc2-jax are comparing it to the libraries listed below
- A benchmark for offline goal-conditioned RL and offline RL☆126Updated this week
- ☆21Updated 6 months ago
- Code for SAPG: Split and Aggregate Policy Gradients (ICML 2024)☆46Updated 5 months ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆66Updated 8 months ago
- ☆58Updated last month
- Skeleton for scalable and flexible Jax RL implementations☆73Updated last year
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆77Updated 9 months ago
- Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)☆17Updated 8 months ago
- (ICLR 2024) Reverse Forward Curriculum Learning☆42Updated 3 months ago
- DrM, a visual RL algorithm, minimizes the dormant ratio to guide exploration-exploitation trade-offs, achieving significant improvements …☆70Updated 9 months ago
- Parallel Q-Learning: Scaling Off-policy Reinforcement Learning under Massively Parallel Simulation☆66Updated last year
- Goal-Conditioned Reinforcement Learning with JAX☆123Updated this week
- Official codebase for "Privileged Sensing Scaffolds Reinforcement Learning", contains the Scaffolder algorithm and Sensory Scaffolding Su…☆24Updated 10 months ago
- [ICLR 2024] Official implementation for "Towards Diverse Behaviors: A Benchmark for Imitation Learning with Human Demonstrations"☆58Updated 2 weeks ago
- ☆260Updated 2 years ago
- ☆74Updated last week
- ☆48Updated 2 years ago
- ☆39Updated 2 months ago
- Code and website for Behavior Transformers: Cloning k modes with one stone.☆116Updated last year
- A minimal and stable PPO.☆131Updated last year
- Source files to replicate experiments in my ICLR 2022 paper.☆69Updated 8 months ago
- Code for "TACO: Temporal Latent Action-Driven Contrastive Loss for Visual Reinforcement Learning"☆23Updated 9 months ago
- JAX implementation of WSRL and RL baselines | ICLR 2025☆30Updated last month
- ☆51Updated 5 months ago
- Skill-based Model-based Reinforcement Learning (CoRL 2022)☆55Updated 2 years ago
- From Imitation to Refinement -- Residual RL for Precise Visual Assembly☆97Updated 3 months ago
- ☆38Updated last month
- A Minimal Example of Isaac Gym with DQN and PPO.☆102Updated last year
- Learning to Walk from Three Minutes of Real-World Data with Semi-structured Dynamics Models☆23Updated 4 months ago