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
- Probabilistic inference for models of behaviour☆11Mar 5, 2026Updated 3 weeks ago
- POPGym Library in JAX☆12Apr 15, 2024Updated last year
- ☆14Oct 7, 2022Updated 3 years ago
- Bayesian model reduction for probabilistic machine learning☆11Jul 3, 2025Updated 8 months ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting with the flexibility to host WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Cloudways by DigitalOcean.
- 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
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 2022Updated 3 years ago
- Repository for the paper: "Curious Exploration via Structured World Models Yields Zero-Shot Object Manipulation" @ NeurIPS 2022☆21Jul 10, 2023Updated 2 years ago
- Fast reinforcement learning research☆61Dec 7, 2024Updated last year
- 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
- Deep Multi-Agent Reinforcement Learning with StarCraft 2☆10Sep 27, 2020Updated 5 years ago
- Simplistic Pytorch Implementation of the Dreamer-RL☆20May 7, 2025Updated 10 months ago
- Robust Reinforcement Learning Benchmark☆12Sep 22, 2024Updated last year
- Code for paper Rethinking the Data Annotation Process for Multi-view 3D Pose Estimation with Active Learning and Self-Training☆22Apr 18, 2023Updated 2 years ago
- Wordpress hosting with auto-scaling on Cloudways • AdFully Managed hosting built for WordPress-powered businesses that need reliable, auto-scalable hosting. Cloudways SafeUpdates now available.
- Bayesian active RL (BARL) and trajectory information planning (TIP)☆26Oct 11, 2022Updated 3 years ago
- (Official) PyTorch implementation for Efficient Episodic Memory Utilization of Cooperative Multi-Agent Reinforcement Learning (EMU) in IC…☆12May 23, 2024Updated last year
- [AAAI 2021 Workshop] The official repository for the LST-MAP model for few-shot image classification.☆13Feb 12, 2021Updated 5 years ago
- Neural Graphical models are neural network based graphical models that offer richer representation, faster inference & sampling☆30Aug 12, 2025Updated 7 months ago
- Evaluating long-term memory of reinforcement learning algorithms☆167Jun 23, 2023Updated 2 years ago
- Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"☆39Jun 11, 2025Updated 9 months ago
- [ICLR 2023] Choreographer: a world-model-based agent that discovers and learns unsupervised skills in latent imagination, and it's able t…☆42Jun 18, 2024Updated last year
- PlaNet: Learning Latent Dynamics for Planning from Pixels☆10Feb 13, 2020Updated 6 years ago
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Jan 7, 2025Updated last year
- 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.
- A simple example of randomized ensembled double q learning☆19Sep 3, 2021Updated 4 years ago
- ☆11Jun 7, 2023Updated 2 years ago
- code for CoRL 2020 paper "Contrastive Variational Model-Based Reinforcement Learning for Complex Observations"☆24Dec 29, 2021Updated 4 years ago
- PyTorch implementation for "Discovery of Incremental Skills" (DISk) algorithm from ICLR 2022 paper "One After Another: Learning Increment…☆20Mar 22, 2022Updated 4 years ago
- An RNN-based reinforcement learning model☆11Jun 29, 2023Updated 2 years ago
- Implementation of 6 DQN extension methods using Pytorch. (RAINBOW)☆16Dec 7, 2020Updated 5 years ago
- Implementation of Neurips 2023 Paper "Multi Time Scale World Models"☆16Nov 8, 2024Updated last year
- 🧶 Minimal PyTorch Soft Actor Critic (SAC) implementation☆38Feb 19, 2022Updated 4 years ago
- 🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)☆18Jul 6, 2023Updated 2 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Implementation of VQ-VAE with a GPT-style sampler in the JAX and Haiku ecosystem.☆12Nov 23, 2023Updated 2 years ago
- SGD (stochastic gradient descent) with OGR - online gradient regression Hessian estimator☆16May 11, 2023Updated 2 years ago
- Minimal A2C/A3C example of an LSTM-based meta-learner.☆13Feb 2, 2021Updated 5 years ago
- Experiment utility code, specifically designed for use with Compute Canada.☆11Jan 27, 2025Updated last year
- Code for the paper "Harnessing Discrete Representations for Continual Reinforcement Learning"☆15Jun 16, 2024Updated last year
- Minimizing Control for Credit Assignment with Strong Feedback☆14Nov 3, 2024Updated last year
- Extending rllab to event-driven multiagent environments☆13Oct 1, 2018Updated 7 years ago