Dreamer 4 jax implementation
☆106Jul 24, 2026Updated 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Unofficial implementation of the Dreamer 4 world model in PyTorch.☆417Jul 9, 2026Updated 3 months ago
- Code, results and checkpoints for the paper "Silent Pipeline Defects in World-Model RL: A Dreamer 4 Case Study": a PyTorch reimplementati…☆39Updated this week
- A simple, performant and scalable JAX-based world modeling codebase.☆165Jan 15, 2026Updated 9 months ago
- Implementation of Danijar's latest iteration for his Dreamer line of work☆219Updated this week
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆107Jan 23, 2025Updated last year
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Official code repository for the paper "Hallucination in World Models is Predictable and Preventable".☆117Oct 20, 2026Updated last week
- ☆83Oct 29, 2025Updated last year
- ☆34Jun 21, 2024Updated 2 years ago
- Implementation of Dreamer v3 in pytorch.☆888Mar 8, 2026Updated 7 months ago
- ☆12May 21, 2026Updated 5 months ago
- Open-source Dreamer world-model implementation in JAX☆397Aug 5, 2026Updated 2 months ago
- Official code repository for the paper "Learning Massively Multitask World Models for Continuous Control".☆140Jan 9, 2026Updated 9 months ago
- ☆58Jan 20, 2023Updated 3 years ago
- Code for the paper Normalizing Flows are Capable Models for RL☆20Jun 3, 2025Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Scalable metrics logging and analysis☆20Updated this week
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.☆289Sep 21, 2026Updated last month
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.☆16Aug 1, 2022Updated 4 years ago
- RL and MARL with Flax NNX☆16Sep 7, 2026Updated last month
- Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"☆993Oct 20, 2026Updated last week
- Decoupled Q-Chunking☆75May 3, 2026Updated 5 months ago
- Mastering Diverse Domains through World Models☆3,917Oct 4, 2026Updated 3 weeks ago
- Partially Observable Benchmarks in JAX☆29Apr 30, 2026Updated 6 months ago
- Simplest and Cleanest DreamerV3 implementation out there☆180Mar 18, 2025Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Accelerated minigrid environments with JAX☆184Sep 10, 2026Updated last month
- A framework for Reinforcement Learning research.☆298Oct 19, 2026Updated last week
- Pytorch version of Dreamer, which follows the original TF v2 codes.☆143Feb 7, 2022Updated 4 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆58May 21, 2023Updated 3 years ago
- Simple Droid platform stack☆48Aug 19, 2026Updated 2 months ago
- ☆17Apr 14, 2026Updated 6 months ago
- [ICLR 2025] Learning Transformer-based World Models with Contrastive Predictive Coding (TWISTER)☆60Mar 9, 2025Updated last year
- A benchmark for offline goal-conditioned RL and offline RL☆511Updated this week
- Building blocks for productive research☆73Oct 4, 2026Updated 3 weeks ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- PyTorch implementation of DreamerV2 model-based RL algorithm☆239Apr 26, 2023Updated 3 years ago
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 11 months ago
- Really Fast End-to-End Jax RL Implementations☆1,119Sep 9, 2024Updated 2 years ago
- ExORL: Exploratory Data for Offline Reinforcement Learning☆140Feb 8, 2022Updated 4 years ago
- A comprehensive JAX/NNX library for diffusion and flow matching generative algorithms, featuring DiT (Diffusion Transformer) and its vari…☆156Oct 16, 2025Updated last year
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆283Jun 10, 2026Updated 4 months ago
- GCRL in JAX. Official repository for LEO (ICML 2026).☆35Jun 20, 2026Updated 4 months ago