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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆53Jan 20, 2023Updated 3 years ago
- Corax: Core RL in JAX☆41Feb 22, 2024Updated 2 years ago
- JAX implementations of core Deep RL algorithms☆84May 2, 2022Updated 3 years ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- Standalone library of frequently-used wrappers for dm_env environments.☆19Jul 9, 2024Updated last year
- ☆278Aug 20, 2022Updated 3 years ago
- Optim4RL is a Jax framework of learning to optimize for reinforcement learning.☆28Nov 27, 2024Updated last year
- PAIRED in PyTorch 🔥☆64Mar 8, 2023Updated 3 years ago
- Codes for "Efficient Offline Policy Optimization with a Learned Model", ICLR2023☆30Jul 18, 2023Updated 2 years ago
- Code for "Planning Goals for Exploration", ICLR2023 Spotlight. An unsupervised RL agent for hard exploration tasks.☆83May 13, 2024Updated last year
- 🔍 Codebase for the ICML '20 paper "Ready Policy One: World Building Through Active Learning" (arxiv: 2002.02693)☆18Jul 6, 2023Updated 2 years ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆753Oct 26, 2022Updated 3 years ago
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆105May 17, 2022Updated 3 years ago
- Bipedal Skills Benchmark for Reinforcement Learning☆25Oct 27, 2022Updated 3 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- An implementation of MuZero in JAX.☆57Nov 8, 2022Updated 3 years ago
- Fast reinforcement learning research☆61Dec 7, 2024Updated last year
- ☆14Jun 17, 2022Updated 3 years ago
- Solving reinforcement learning tasks which require language and vision☆33Apr 4, 2023Updated 2 years ago
- CIC: Contrastive Intrinsic Control for Unsupervised Skill Discovery☆85Jul 27, 2022Updated 3 years ago
- ☆18Updated this week
- This repository contains the implementation of the PTR algorithm described in the paper: Pre-Training for Robots: Leveraging Diverse Mult…☆32Oct 26, 2022Updated 3 years ago
- Variational Reinforcement Learning☆17Jul 25, 2024Updated last year
- RL Environments in JAX 🌍☆871May 30, 2025Updated 9 months ago
- Evaluation of TD-MPC2.☆21Jan 21, 2024Updated 2 years ago
- Vectorization techniques for fast population-based training.☆57Aug 12, 2022Updated 3 years ago
- Accelerated replay buffers in JAX☆46Sep 17, 2022Updated 3 years ago
- Offline RL experiments☆15Oct 1, 2022Updated 3 years ago
- Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.☆138Aug 20, 2024Updated last year
- Modular Single-file Reinfocement Learning Algorithms Library☆38May 16, 2023Updated 2 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆122Aug 22, 2024Updated last year
- JAX code for the paper "Control-Oriented Model-Based Reinforcement Learning with Implicit Differentiation"☆44Jun 14, 2021Updated 4 years ago
- ROBEL: Robotics Benchmarks for Learning with low-cost robots☆148Nov 5, 2020Updated 5 years ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆868Aug 12, 2024Updated last year
- Experiment. Plot. Tabulate.☆73Aug 22, 2024Updated last year
- 🪐 The Sebulba architecture to scale reinforcement learning on Cloud TPUs in JAX☆61Oct 23, 2023Updated 2 years ago
- Co-Adaptation of Algorithmic and Implementational Innovations in Inference-based Deep Reinforcement Learning (NeurIPS2021)☆20Oct 25, 2021Updated 4 years ago
- ☆234Sep 3, 2023Updated 2 years ago
- Reinforcement learning library for PyTorch.☆11Jun 15, 2018Updated 7 years ago