Dreamer on JAX
☆16Jan 19, 2022Updated 4 years ago
Alternatives and similar repositories for jax-dreamer
Users that are interested in jax-dreamer are comparing it to the libraries listed below
Sorting:
- A scalable Dreamer implementation in JAX☆10May 22, 2022Updated 3 years ago
- On the Feasibility of Cross-Task Transfer with Model-Based Reinforcement Learning☆16Apr 30, 2023Updated 2 years ago
- Temporally Correlated Episodic Reinforcement Learning, ICLR 24☆12Apr 8, 2024Updated last year
- Code for "Theoretical Foundations of Deep Selective State-Space Models" (NeurIPS 2024)☆15Jan 7, 2025Updated last year
- Codebase for the paper "How Crucial is Transformer in Decision Transformer?". Containing experiments on different pendulum tasks and code…☆28Mar 24, 2023Updated 2 years ago
- LAMBDA is a model-based reinforcement learning agent that uses Bayesian world models for safe policy optimization☆38Jan 16, 2023Updated 3 years ago
- A short conceptual replication of "Prefrontal cortex as a meta-reinforcement learning system" in Jax.☆18Feb 27, 2023Updated 3 years ago
- ☆19Mar 1, 2023Updated 3 years ago
- RAD: Reinforcement Learning with Augmented Data (code for procgen experiments)☆18Mar 29, 2021Updated 4 years ago
- CREATE Environment for long-horizon physics-puzzle tasks with diverse tools☆18Nov 22, 2022Updated 3 years ago
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆22Oct 12, 2023Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 2022Updated 3 years ago
- ☆91Sep 19, 2022Updated 3 years ago
- A port of muP to JAX/Haiku☆25Oct 23, 2022Updated 3 years ago
- Fast dataset format and loader☆24Jan 2, 2026Updated 2 months ago
- Simple single file implementations of Reinforcement Learning algorithms in Julia☆23Feb 15, 2025Updated last year
- speed-running solving robot manipulation tasks☆24Oct 31, 2024Updated last year
- ☆23Aug 19, 2022Updated 3 years ago
- Official Code for "Relative Entropy Pathwise Policy Optimization"☆46Feb 27, 2026Updated last week
- Non official implementation of the Linear Recurrent Unit (LRU, Orvieto et al. 2023)☆62Sep 3, 2025Updated 6 months ago
- This repository contains the implementation of the PTR algorithm described in the paper: Pre-Training for Robots: Leveraging Diverse Mult…☆30Oct 26, 2022Updated 3 years ago
- Reinforcement Learning inside a 3D soccer simulation☆37Sep 15, 2024Updated last year
- https://x.com/BlinkDL_AI/status/1884768989743882276☆28May 4, 2025Updated 10 months ago
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."☆37Jul 11, 2025Updated 7 months ago
- A simple script to see how my ideas evolve over time☆44Jun 4, 2025Updated 9 months ago
- Open source code for EigenGame.☆35May 15, 2023Updated 2 years ago
- Simple JAX Graphics Library.☆36Nov 3, 2024Updated last year
- ☆35Jan 29, 2023Updated 3 years ago
- Unofficial implementation of Linear Recurrent Units, by Deepmind, in Pytorch☆74Apr 22, 2025Updated 10 months ago
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 2 years ago
- Proximal Policy Optimization (Continuous Version) in PyTorch.☆29May 12, 2025Updated 9 months ago
- The official starter-kit for NeurIPS 2025 mind games competition☆21Jul 27, 2025Updated 7 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆101Jan 23, 2025Updated last year
- MoDem Accelerating Visual Model-Based Reinforcement Learning with Demonstrations☆86Dec 12, 2022Updated 3 years ago
- ☆35Nov 22, 2024Updated last year
- Reinforcement learning pipeline for specific configured robot to learn to reach randomly sampled target poses within its workspace.☆20Aug 28, 2025Updated 6 months ago
- Directed masked autoencoders☆14Feb 20, 2026Updated 2 weeks ago
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆89Nov 19, 2023Updated 2 years ago
- About The dataset was recorded on the Husky robotics platform on the university campus and consists of 5 tracks recorded at different tim…☆11Mar 25, 2025Updated 11 months ago