yardenas / jax-dreamerView external linksLinks
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:
- 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
- On the Feasibility of Cross-Task Transfer with Model-Based Reinforcement Learning☆16Apr 30, 2023Updated 2 years ago
- 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
- Code for the paper "Harnessing Discrete Representations for Continual Reinforcement Learning"☆15Jun 16, 2024Updated last year
- 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 2 years ago
- ☆19Mar 1, 2023Updated 2 years ago
- Learning Robust Dynamics Through Variational Sparse Gating☆20Oct 19, 2022Updated 3 years ago
- Fast dataset format and loader☆23Jan 2, 2026Updated last month
- Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.☆22Oct 12, 2023Updated 2 years ago
- CREATE Environment for long-horizon physics-puzzle tasks with diverse tools☆18Nov 22, 2022Updated 3 years ago
- ☆91Sep 19, 2022Updated 3 years ago
- A port of muP to JAX/Haiku☆25Oct 23, 2022Updated 3 years 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
- Optimal Transport for Machine Learners☆46Nov 15, 2025Updated 3 months ago
- ☆23Aug 19, 2022Updated 3 years ago
- Official Code for "Relative Entropy Pathwise Policy Optimization"☆45Updated this week
- ☆27Jan 8, 2026Updated last month
- Reinforcement Learning inside a 3D soccer simulation☆37Sep 15, 2024Updated last year
- 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
- A simple script to see how my ideas evolve over time☆44Jun 4, 2025Updated 8 months ago
- This code accompanies the paper "Leveraging Skills from Unlabeled Prior Data for Efficient Online Exploration."☆36Jul 11, 2025Updated 7 months ago
- https://x.com/BlinkDL_AI/status/1884768989743882276☆28May 4, 2025Updated 9 months ago
- Open source code for EigenGame.☆34May 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 9 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 6 months ago
- JAX reimplementation of the DeepMind paper "Genie: Generative Interactive Environments"☆100Jan 23, 2025Updated last year
- MoDem Accelerating Visual Model-Based Reinforcement Learning with Demonstrations☆86Dec 12, 2022Updated 3 years ago
- Reinforcement learning pipeline for specific configured robot to learn to reach randomly sampled target poses within its workspace.☆19Aug 28, 2025Updated 5 months ago
- A reinforcement learning package implemented in Torch☆11Jan 24, 2016Updated 10 years ago
- ☆15Jul 27, 2023Updated 2 years ago
- Public code for "Reinforcement Learning from Passive Data via Latent Intentions"☆89Nov 19, 2023Updated 2 years ago
- Partially Observable Multi-Agent RL with Transformers☆17Updated this week