Building blocks for productive research
☆69Jan 12, 2026Updated last month
Alternatives and similar repositories for elements
Users that are interested in elements are comparing it to the libraries listed below
Sorting:
- General Modules for JAX☆72Feb 21, 2026Updated last week
- Fast and reliable distributed systems in Python☆33Jan 12, 2026Updated last month
- Fast reinforcement learning research☆61Dec 7, 2024Updated last year
- Standalone library of frequently-used wrappers for dm_env environments.☆18Jul 9, 2024Updated last year
- Efficiently send large arrays across machines☆17Jul 24, 2024Updated last year
- Fast dataset format and loader☆23Jan 2, 2026Updated last month
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 years ago
- Scalable metrics logging and analysis☆18Apr 9, 2025Updated 10 months ago
- Tools and Utils for Experiments (TUX)☆15Jan 28, 2025Updated last year
- JAX implementation of deep RL agents with resets from the paper "The Primacy Bias in Deep Reinforcement Learning"☆104May 17, 2022Updated 3 years ago
- This repository is a collection of widely used self-supervised auxiliary losses used for learning representations in reinforcement learni…☆14Feb 27, 2023Updated 3 years ago
- Implementation of CASCADE in Learning General World Models in a Handful of Reward-Free Deployments (NeurIPS 22).☆29Oct 25, 2022Updated 3 years ago
- [deprecated] Engine Agnostic Gym Environment for Robotics☆17Feb 10, 2022Updated 4 years ago
- Synchronized Curriculum Learning for RL Agents☆122Feb 21, 2026Updated last week
- [CoRL 2021] A robotics benchmark for cross-embodiment imitation.☆60Oct 4, 2023Updated 2 years ago
- Deep Hierarchical Planning from Pixels☆114Dec 21, 2022Updated 3 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆164Jun 23, 2023Updated 2 years ago
- An ML research codebase built with friends :)☆26Aug 25, 2024Updated last year
- Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)☆22Jul 21, 2025Updated 7 months ago
- Bayesian active RL (BARL) and trajectory information planning (TIP)☆26Oct 11, 2022Updated 3 years ago
- ☆10Jun 27, 2024Updated last year
- A tiny reinforcement learning codebase for continuous control, built on top of JAX.☆15Mar 28, 2023Updated 2 years ago
- ☆25Sep 23, 2024Updated last year
- A repository for the updated version of CoinRun used to collect MUGEN, a multimodal video-audio-text dataset. This repo contains scripts …☆13Jul 13, 2022Updated 3 years ago
- Accompanying repository for Unsupervised Active Domain Randomization in Goal-Directed RL☆12Aug 4, 2020Updated 5 years ago
- Propose & vote on reading group papers in the "Discussions" tab.☆12Feb 20, 2024Updated 2 years ago
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆37May 19, 2023Updated 2 years ago
- ☆59Sep 22, 2022Updated 3 years ago
- MuJoCo models for Unitree Robots☆12Nov 24, 2021Updated 4 years ago
- Docker containers of baseline agents for the Crafter environment☆30Dec 14, 2021Updated 4 years ago
- Deep Reinforcement Learning by using an on-policy adaptation of Maximum a Posteriori Policy Optimization (MPO)☆16Oct 23, 2021Updated 4 years ago
- Reinforcement learning in pure JAX.☆13Dec 24, 2025Updated 2 months ago
- A GPU-Accelerated Mahjong Simulator for RL in JAX☆23Feb 12, 2026Updated 2 weeks ago
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆13Jul 19, 2024Updated last year
- Official code release for "CrossQ: Batch Normalization in Deep Reinforcement Learning for Greater Sample Efficiency and Simplicity"☆88Jun 4, 2024Updated last year
- ☆160Feb 18, 2026Updated last week
- ☆88Jan 9, 2026Updated last month
- Code for the paper "Learning to Assist Humans without Inferring Rewards"☆19Jul 7, 2024Updated last year
- ☆17Feb 12, 2025Updated last year