Building blocks for productive research
☆69Jan 12, 2026Updated 2 months ago
Alternatives and similar repositories for elements
Users that are interested in elements are comparing it to the libraries listed below
Sorting:
- Fast and reliable distributed systems in Python☆33Jan 12, 2026Updated 2 months ago
- General Modules for JAX☆72Feb 21, 2026Updated last month
- Fast reinforcement learning research☆61Dec 7, 2024Updated last year
- Standalone library of frequently-used wrappers for dm_env environments.☆19Jul 9, 2024Updated last year
- Efficiently send large arrays across machines☆17Jul 24, 2024Updated last year
- Scalable metrics logging and analysis☆18Apr 9, 2025Updated 11 months ago
- Fast dataset format and loader☆24Mar 6, 2026Updated 2 weeks ago
- Clockwork VAEs in JAX/Flax☆32Jul 16, 2021Updated 4 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
- Implementation of CASCADE in Learning General World Models in a Handful of Reward-Free Deployments (NeurIPS 22).☆29Oct 25, 2022Updated 3 years ago
- Synchronized Curriculum Learning for RL Agents☆122Mar 7, 2026Updated 2 weeks ago
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.☆17Aug 1, 2022Updated 3 years ago
- A tiny reinforcement learning codebase for continuous control, built on top of JAX.☆15Mar 28, 2023Updated 2 years ago
- 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
- Tools and Utils for Experiments (TUX)☆15Jan 28, 2025Updated last year
- ☆18Mar 10, 2026Updated last week
- Evaluating long-term memory of reinforcement learning algorithms☆165Jun 23, 2023Updated 2 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
- [deprecated] Engine Agnostic Gym Environment for Robotics☆17Feb 10, 2022Updated 4 years ago
- ☆59Sep 22, 2022Updated 3 years ago
- Official code for TLDR: Unsupervised Goal-Conditioned RL via Temporal Distance-Aware Representations☆37Jan 24, 2026Updated last month
- Deep Hierarchical Planning from Pixels☆118Dec 21, 2022Updated 3 years ago
- ☆10Jun 27, 2024Updated last year
- Official release of the DMControl Generalization Benchmark 2 (DMC-GB2)☆22Jul 21, 2025Updated 8 months ago
- [RA-L 2024] Novel action spaces leveraging redundancy in 7 DoF arms enable efficient & precise learning in robotic manipulation☆21Jun 6, 2024Updated last year
- [CoRL 2021] A robotics benchmark for cross-embodiment imitation.☆60Oct 4, 2023Updated 2 years ago
- ☆160Feb 18, 2026Updated last month
- Standardized Minecraft Diamond Environment for Reinforcement Learning☆37May 19, 2023Updated 2 years ago
- LeanRL is a fork of CleanRL, where selected PyTorch scripts optimized for performance using compile and cudagraphs.☆680Aug 22, 2025Updated 6 months ago
- Challenges and Opportunities in Offline Reinforcement Learning from Visual Observations☆112May 27, 2024Updated last year
- A GPU-Accelerated Mahjong Simulator for RL in JAX☆23Feb 12, 2026Updated last month
- [CoRL 2022] BusyBot: Learning to Interact, Reason, and Plan in a BusyBoard Environment☆19May 17, 2023Updated 2 years ago
- Deep Reinforcement Learning by using an on-policy adaptation of Maximum a Posteriori Policy Optimization (MPO)☆16Oct 23, 2021Updated 4 years ago
- Code release for "Training Robots to Evaluate Robots" (CoRL'22, Best Paper Award)☆17Feb 15, 2023Updated 3 years ago
- Revisiting Peng's Q(lambda) for Modern Reinforcement Learning☆15Jul 23, 2021Updated 4 years ago
- Jaxplorer is a Jax reinforcement learning (RL) framework for exploring new ideas.☆13Jul 19, 2024Updated last year
- Code for MOBILE: Model-Bellman Inconsistency Penalized Offline Policy Optimization☆23Apr 17, 2024Updated last year
- ☆25Sep 23, 2024Updated last year
- ☆18May 30, 2023Updated 2 years ago