ShaneFlandermeyer / tdmpc2-jax
Jax/Flax Implementation of TD-MPC2
☆61Updated 2 weeks ago
Alternatives and similar repositories for tdmpc2-jax
Users that are interested in tdmpc2-jax are comparing it to the libraries listed below
Sorting:
- JAX implementation of WSRL and RL baselines | ICLR 2025☆42Updated 3 weeks ago
- [ICLR 2025] Bootstrapped Model Predictive Control☆13Updated 3 weeks ago
- Official repo for paper "TD-M(PC)^2: Improving Temporal Difference MPC Through Policy Constraint"☆55Updated 3 months ago
- ☆91Updated 2 weeks ago
- Parallel Q-Learning: Scaling Off-policy Reinforcement Learning under Massively Parallel Simulation☆70Updated last year
- A benchmark for offline goal-conditioned RL and offline RL☆165Updated last month
- Code for SAPG: Split and Aggregate Policy Gradients (ICML 2024)☆48Updated 8 months ago
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆73Updated 11 months ago
- Official codebase for "Privileged Sensing Scaffolds Reinforcement Learning", contains the Scaffolder algorithm and Sensory Scaffolding Su…☆27Updated last year
- (ICLR 2024) Reverse Forward Curriculum Learning☆47Updated 5 months ago
- Goal-Conditioned Reinforcement Learning with JAX☆154Updated this week
- ☆24Updated 8 months ago
- Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)☆19Updated 11 months ago
- Unified Implementations of Offline Reinforcement Learning Algorithms☆71Updated 2 weeks ago
- Source files to replicate experiments in my ICLR 2022 paper.☆71Updated 10 months ago
- Skeleton for scalable and flexible Jax RL implementations☆81Updated last year
- [ICLR 2024] Official implementation for "Towards Diverse Behaviors: A Benchmark for Imitation Learning with Human Demonstrations"☆77Updated 3 months ago
- ☆85Updated 2 months ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆77Updated last year
- A minimal and stable PPO.☆136Updated last year
- HIQL: Offline Goal-Conditioned RL with Latent States as Actions (NeurIPS 2023)☆82Updated 5 months ago
- PWM: Policy Learning with Large World Models☆46Updated 2 months ago
- Learning to Walk from Three Minutes of Real-World Data with Semi-structured Dynamics Models☆30Updated 6 months ago
- ☆43Updated 4 months ago
- Code of the paper "LS-IQ: Implicit Reward Regularization for Inverse Reinforcement Learning" & LocoMuJoCo Baselines☆52Updated last year
- Wrappers and utilities for Nvidia IsaacGym☆98Updated 3 years ago
- Challenging dexterous manipulation environments for RL that extend the hand manipulation environments introduced in OpenAI's Gym☆45Updated 3 years ago
- Safe Multi-Agent Isaac Gym benchmark for safe multi-agent reinforcement learning research.☆61Updated 4 months ago
- Rex is a JAX-powered framework for sim-to-real robotics.☆39Updated 3 months ago
- From Imitation to Refinement -- Residual RL for Precise Visual Assembly☆117Updated 5 months ago