Dreamer 4 jax implementation
☆75Nov 28, 2025Updated 3 months ago
Alternatives and similar repositories for dreamer4-jax
Users that are interested in dreamer4-jax are comparing it to the libraries listed below
Sorting:
- ☆32Jun 21, 2024Updated last year
- ☆13Dec 14, 2024Updated last year
- Unofficial implementation of the Dreamer 4 world model in PyTorch.☆247Jan 18, 2026Updated 2 months ago
- Minimal Transformer base in JAX. A single backbone for language modelling, diffusion, classification, etc...☆14May 28, 2025Updated 9 months ago
- Implementations of Temporal Difference InfoNCE (TD InfoNCE)☆34Nov 13, 2023Updated 2 years ago
- Resilient Model-Based RL by Regularizing Posterior Predictability☆22Mar 4, 2024Updated 2 years ago
- ☆10Apr 13, 2023Updated 2 years ago
- Reinforcement learning with Equinox☆20Mar 4, 2025Updated last year
- Asymmetric methods for partially observable reinforcement learning☆10Jun 9, 2025Updated 9 months ago
- Official implementation for the paper: "Real-Time Inverse Kinematics for Generating Multi-Constrained Movements of Virtual Human Characte…☆18Feb 19, 2026Updated last month
- Environments for Active Vision Reinforcement Learning☆29Oct 10, 2024Updated last year
- ViViDex implementation under the MuJoCo simulator, ICRA 2025☆18Apr 9, 2025Updated 11 months ago
- Code release for "HarmonyDream: Task Harmonization Inside World Models" (ICML 2024), https://arxiv.org/abs/2310.00344☆51Jun 27, 2024Updated last year
- Code for Scalable Offline Model-Based RL with Action chunking☆20Feb 20, 2026Updated last month
- Code for the paper Normalizing Flows are Capable Models for RL☆18Jun 3, 2025Updated 9 months ago
- High quality implementations of imitation and inverse reinforcement learning algorithms☆24Aug 19, 2025Updated 7 months ago
- ReDiffuser: Reliable Decision-Making Using a Diffuser with Confidence Estimation☆15Jun 2, 2024Updated last year
- DiWA: Diffusion Policy Adaptation with World Models☆74Mar 4, 2026Updated 2 weeks ago
- Official code for ACT: Empowering Decision Transformer with Dynamic Programming via Advantage Conditioning (AAAI'24)☆17Feb 10, 2024Updated 2 years ago
- The repository for paper "PIB: Priority-Based Information Bottleneck for Collaborative Edge Inference in Video Analytics".☆15Mar 23, 2025Updated 11 months ago
- This repository contains code examples for the paper "Learning to sequence and blend robotics skills via differentiable optimization".☆12Sep 11, 2022Updated 3 years ago
- This repository contains the implementation of all methods evaluated in the paper "Learning a Thousand Tasks in a Day". We provide model …☆139Nov 17, 2025Updated 4 months ago
- A benchmark for evaluating reinforcement learning algorithms that train the policies using imaginary rollouts from LLMs.☆14Nov 4, 2025Updated 4 months ago
- MENTOR is a highly efficient visual RL algorithm that excels in both simulation and real-world complex robotic learning tasks.☆27Jul 9, 2025Updated 8 months ago
- code for the paper Offline Prioritized Experience Replay☆12Jun 13, 2023Updated 2 years ago
- ☆11Oct 3, 2022Updated 3 years ago
- ☆28Jan 8, 2026Updated 2 months ago
- Graph-Assisted Stitching for Offline Hierarchical Reinforcement Learning (ICML 2025)☆17Dec 9, 2025Updated 3 months ago
- Implementation of Dreamer v3 in pytorch.☆820Mar 8, 2026Updated last week
- Unbiased and Consistent Nested Sampling via Sequential Monte Carlo☆15May 4, 2025Updated 10 months ago
- ☆17Jun 4, 2021Updated 4 years ago
- ☆22May 12, 2025Updated 10 months ago
- LUMOS: Language-Conditioned Imitation Learning with World Models☆16Mar 4, 2026Updated 2 weeks ago
- Implementation of Neurips 2023 Paper "Multi Time Scale World Models"☆16Nov 8, 2024Updated last year
- ☆13Sep 24, 2022Updated 3 years ago
- Awesome Goal-Conditioned Reinforcement Learning☆23Feb 22, 2026Updated 3 weeks ago
- Being-M0.5: A Real-Time Controllable Vision-Language-Motion Model (ICCV 2025)☆35Sep 4, 2025Updated 6 months ago
- ☆93May 31, 2025Updated 9 months ago
- Official implementation of Harnessing Mixed Offline Reinforcement Learning Datasets via Trajectory Reweighting☆16Feb 14, 2024Updated 2 years ago