Reinforcement learning library in JAX.
☆101Oct 22, 2023Updated 2 years ago
Alternatives and similar repositories for magi
Users that are interested in magi are comparing it to the libraries listed below
Sorting:
- ☆52Jan 20, 2023Updated 3 years ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- Corax: Core RL in JAX☆39Feb 22, 2024Updated 2 years ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 3 years ago
- ☆279Aug 20, 2022Updated 3 years ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆82May 13, 2024Updated last year
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆105May 17, 2022Updated 3 years ago
- Standalone library of frequently-used wrappers for dm_env environments.☆18Jul 9, 2024Updated last year
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆752Oct 26, 2022Updated 3 years ago
- GPT implementation in Flax☆18Jan 8, 2022Updated 4 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆83Jul 27, 2022Updated 3 years ago
- Fast reinforcement learning research☆61Dec 7, 2024Updated last year
- Bipedal Skills Benchmark for Reinforcement Learning☆25Oct 27, 2022Updated 3 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
- A set of environments utilizing pybullet for simulation of robotic manipulation tasks.☆29Mar 8, 2021Updated 4 years ago
- Optim4RL is a Jax framework of learning to optimize for reinforcement learning.☆28Nov 27, 2024Updated last year
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆30Jul 18, 2023Updated 2 years ago
- Reinforcement learning library for PyTorch.☆11Jun 15, 2018Updated 7 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- An implementation of MuZero in JAX.☆57Nov 8, 2022Updated 3 years ago
- Solving reinforcement learning tasks which require language and vision☆33Apr 4, 2023Updated 2 years ago
- PAIRED in PyTorch 🔥☆64Mar 8, 2023Updated 2 years ago
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 2 years ago
- RL Environments in JAX 🌍☆864May 30, 2025Updated 9 months ago
- Evaluation of TD-MPC2.☆21Jan 21, 2024Updated 2 years ago
- Variational Reinforcement Learning☆17Jul 25, 2024Updated last year
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆61Oct 23, 2023Updated 2 years ago
- A simple wrapper to analyse and visualise reinforcement learning agents' behaviour in the environment.☆14Jan 8, 2022Updated 4 years ago
- Implementation for "ROLL: Visual Self-Supervised Reinforcement Learning with Object Reasoning", CoRL 2020☆16Jun 22, 2022Updated 3 years ago
- Repository for ML Reproducibility Challenge 2020 for the Neurips paper, "The Value Equivalence Principle for Model-Based Reinforcement Le…☆18Apr 13, 2021Updated 4 years ago
- Open source code combining implementations of Upside Down Reinforcement Learning and Reward Conditioned Policies☆19Mar 10, 2021Updated 4 years ago
- This is a miniature race car gym-env for RL from states (and images)☆28Nov 3, 2021Updated 4 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆122Aug 22, 2024Updated last year
- Experiment. Plot. Tabulate.☆73Aug 22, 2024Updated last year
- A tool for recording RL trajectories.☆111Jul 29, 2025Updated 7 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
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆44Jun 14, 2021Updated 4 years ago
- [CoRL 2021] A robotics benchmark for cross-embodiment imitation.☆60Oct 4, 2023Updated 2 years ago