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 11 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
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆25Feb 15, 2025Updated last year
- Probabilistic inference for models of behaviour☆13Mar 5, 2026Updated 7 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Learning Robust Dynamics Through Variational Sparse Gating☆21Oct 19, 2022Updated 4 years ago
- Mitigating Partial Observability in Sequential Decision Processes via the Lambda Discrepancy☆24Oct 28, 2024Updated 2 years ago
- ☆13Apr 25, 2024Updated 2 years ago
- Fast reinforcement learning research☆65May 25, 2026Updated 5 months ago
- Bayesian model reduction for probabilistic machine learning☆12Jul 3, 2025Updated last year
- ☆15Oct 7, 2022Updated 4 years ago
- A PyTorch Implementation of PlaNet: A Deep Planning Network for Reinforcement Learning☆13Aug 31, 2020Updated 6 years ago
- Minimal Decision Transformer Implementation written in Jax (Flax).☆18Aug 8, 2022Updated 4 years ago
- Agar.io for Continual Reinforcement Learning☆28Aug 27, 2026Updated 2 months ago
- 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.
- Simple JAX Graphics Library.☆39Nov 3, 2024Updated 2 years ago
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆20Feb 27, 2023Updated 3 years ago
- A minimal implementation of Go-Explore without domain knowledge☆15Apr 26, 2021Updated 5 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆185Jun 23, 2023Updated 3 years ago
- ☆29Jul 9, 2024Updated 2 years ago
- Implementation of the Fast Efficient Hyperparameter Tuning for Policy Gradient Methods https://arxiv.org/abs/1902.06583☆19Oct 22, 2019Updated 7 years ago
- CREATE Environment for long-horizon physics-puzzle tasks with diverse tools☆18Nov 22, 2022Updated 3 years ago
- Code for NeurIPS 2022 paper Exploiting Reward Shifting in Value-Based Deep RL☆29Oct 29, 2023Updated 3 years ago
- Accelerated replay buffers in JAX☆47Sep 17, 2022Updated 4 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- 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
- Dreamer on JAX☆16Jan 19, 2022Updated 4 years ago
- POPGym Library in JAX☆14Apr 15, 2024Updated 2 years ago
- Skeleton for scalable and flexible Jax RL implementations☆102Jul 1, 2023Updated 3 years ago
- 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 research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆425Mar 18, 2026Updated 7 months ago
- Repository for the paper: "Curious Exploration via Structured World Models Yields Zero-Shot Object Manipulation" @ NeurIPS 2022☆24Jul 10, 2023Updated 3 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
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- Deep Multi-Agent Reinforcement Learning with StarCraft 2☆10Sep 27, 2020Updated 6 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated last year
- A minimal home grid world environment to evaluate language understanding in interactive agents.☆25Sep 6, 2023Updated 3 years ago
- Official implementation of the paper "Arbitrarily Scalable Environment Generators via Neural Cellular Automata", published in NeurIPS 202…☆13Oct 31, 2023Updated 3 years ago
- Toy meta-RL environments for testing algorithms implementations☆10Feb 21, 2024Updated 2 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
- 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