An implementation of DreamerV2 written in JAX, with support for running multiple random seeds of an experiment on a single GPU.
☆18Jan 16, 2023Updated 3 years ago
Alternatives and similar repositories for dreamerv2_JAX
Users that are interested in dreamerv2_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:
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- Flax Implementation of DreamerV3 on Crafter☆19Nov 29, 2025Updated 9 months ago
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 3 years ago
- Probabilistic inference for models of behaviour☆13Mar 5, 2026Updated 5 months ago
- Learning Robust Dynamics Through Variational Sparse Gating☆21Oct 19, 2022Updated 3 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆24Oct 28, 2024Updated last year
- ☆13Apr 25, 2024Updated 2 years ago
- POPGym Library in JAX☆14Apr 15, 2024Updated 2 years ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆24Feb 15, 2025Updated last year
- Fast reinforcement learning research☆65May 25, 2026Updated 3 months ago
- Agar.io for Continual Reinforcement Learning☆28Aug 27, 2026Updated last week
- Minimal Decision Transformer Implementation written in Jax (Flax).☆18Aug 8, 2022Updated 4 years ago
- Simple JAX Graphics Library.☆38Nov 3, 2024Updated last year
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆13Aug 31, 2020Updated 6 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆21Mar 22, 2022Updated 4 years ago
- A minimal implementation of Go-Explore without domain knowledge☆15Apr 26, 2021Updated 5 years ago
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated last year
- ☆15Oct 7, 2022Updated 3 years ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆422Mar 18, 2026Updated 5 months ago
- ☆29Jul 9, 2024Updated 2 years ago
- Accelerated replay buffers in JAX☆47Sep 17, 2022Updated 3 years ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆29Oct 29, 2023Updated 2 years ago
- Implementation of the Fast Efficient Hyperparameter Tuning for Policy Gradient Methods https://arxiv.org/abs/1902.06583☆19Oct 22, 2019Updated 6 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Pytorch2Jax is a small Python library that provides functions that wraps PyTorch models into Jax functions and Flax modules.☆21Feb 20, 2023Updated 3 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆183Jun 23, 2023Updated 3 years ago
- Repository for the paper: "Curious Exploration via Structured World Models Yields Zero-Shot Object Manipulation" @ NeurIPS 2022☆21Jul 10, 2023Updated 3 years ago
- Dreamer on JAX☆16Jan 19, 2022Updated 4 years ago
- [ICLR 2023] Choreographer: a world-model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able t…☆43Jun 18, 2024Updated 2 years ago
- Implementation of Proximal Policy Optimization in Jax+Flax☆21May 18, 2023Updated 3 years ago
- A videogame made with PyGame turned into an Open AI Gym Learning Environment for Reinforcement Learning agents.☆14Jan 3, 2023Updated 3 years ago
- Deep Multi-Agent Reinforcement Learning with StarCraft 2☆10Sep 27, 2020Updated 5 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- MIMEx: Intrinsic Rewards from Masked Input Modeling [NeurIPS 2023]☆16May 17, 2023Updated 3 years ago
- Novelty MiniGrid--NovGrid--is an extension of MiniGrid environment that allows for the world properties and dynamics to change according …☆34May 21, 2024Updated 2 years ago
- ☆16Jul 16, 2024Updated 2 years ago
- Logging library for JAX that is compatible with transformations and primitives such as vmap and scan.☆16Updated this week
- ☆66Jan 30, 2026Updated 7 months ago
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆20Feb 27, 2023Updated 3 years ago
- Single-file SAC-N implementation on jax with flax and equinox. 10x faster than pytorch☆57May 21, 2023Updated 3 years ago