Dreamer 4 jax implementation
☆101Jul 24, 2026Updated 2 weeks 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.☆380Jul 9, 2026Updated last month
- A simple, performant and scalable JAX-based world modeling codebase.☆157Jan 15, 2026Updated 6 months ago
- From-scratch PyTorch reproduction of DreamerV4 (Hafner et al., 2025): masked-autoencoder tokenizer, block-causal flow-matching dynamics w…☆37Updated this week
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆105Jan 23, 2025Updated last year
- Implementation of Danijar's latest iteration for his Dreamer line of work☆209Jul 4, 2026Updated last month
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- ☆60Jul 13, 2026Updated 3 weeks ago
- ☆34Jun 21, 2024Updated 2 years ago
- Implementation of Dreamer v3 in pytorch.☆887Mar 8, 2026Updated 5 months ago
- ☆58Jan 20, 2023Updated 3 years ago
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.☆17Aug 1, 2022Updated 4 years ago
- Code for the paper Normalizing Flows are Capable Models for RL☆20Jun 3, 2025Updated last year
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆57May 21, 2023Updated 3 years ago
- Official code repository for the paper "Learning Massively Multitask World Models for Continuous Control".☆133Jan 9, 2026Updated 7 months ago
- A framework for Reinforcement Learning research.☆273Jul 28, 2026Updated last week
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- Online Goal-Conditioned Reinforcement Learning in JAX. ICLR 2025 Spotlight.☆276Updated this week
- Decoupled Q-Chunking☆74May 3, 2026Updated 3 months ago
- Accelerated minigrid environments with JAX☆175Oct 20, 2025Updated 9 months ago
- [ICLR 2025] Learning Transformer-based World Models with Contrastive Predictive Coding (TWISTER)☆60Mar 9, 2025Updated last year
- ☆16Apr 14, 2026Updated 3 months ago
- ☆77Oct 29, 2025Updated 9 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆274Jun 10, 2026Updated 2 months ago
- Code for Scalable Offline Model-Based RL with Action chunking☆30Feb 20, 2026Updated 5 months ago
- Mastering Diverse Domains through World Models☆3,647May 25, 2026Updated 2 months ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- A benchmark for offline goal-conditioned RL and offline RL☆459Jan 14, 2026Updated 6 months ago
- PyTorch code accompanying the paper "Imitating Graph-Based Planning with Goal-Conditioned Policies" (ICLR 2023).☆21Mar 4, 2023Updated 3 years ago
- PyTorch implementation of DreamerV2 model-based RL algorithm☆240Apr 26, 2023Updated 3 years ago
- Pointax: PointMaze Environment for JAX☆28Oct 22, 2025Updated 9 months ago
- A scalable benchmark for state representation learning in visual reinforcement learning.☆17Jun 23, 2025Updated last year
- A dataloader, but for JAX☆20May 17, 2024Updated 2 years ago
- Simplest and Cleanest DreamerV3 implementation out there☆172Mar 18, 2025Updated last year
- Building blocks for productive research☆72Mar 26, 2026Updated 4 months ago
- Official repo for paper "TD-M(PC)^2: Improving Temporal Difference MPC Through Policy Constraint"☆88Feb 11, 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.
- Clean single-file implementation of offline RL algorithms in JAX☆182Jun 5, 2026Updated 2 months ago
- Scalable metrics logging and analysis☆18Jun 30, 2026Updated last month
- Code for "TD-MPC2: Scalable, Robust World Models for Continuous Control"☆920Jul 13, 2026Updated 3 weeks ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆280Sep 22, 2025Updated 10 months ago
- DiT (training + flow matching) in Jax☆12Jan 5, 2025Updated last year
- Official code repository for the paper "Hallucination in World Models is Predictable and Preventable".☆90Jul 14, 2026Updated 3 weeks ago
- ☆16Jul 19, 2026Updated 3 weeks ago