RoboEden / jux
JUX is a JAX-accelerated engine for Lux AI Challenge 2022.
☆20Updated last year
Alternatives and similar repositories for jux:
Users that are interested in jux are comparing it to the libraries listed below
- ☆47Updated 2 years ago
- Baselines for gymnax 🤖☆66Updated 2 years ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆111Updated 8 months ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆201Updated 3 weeks ago
- ☆106Updated 3 years ago
- ☆75Updated 5 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆156Updated last month
- megastep helps you build 1-million FPS reinforcement learning environments on a single GPU☆135Updated 3 years ago
- Clean single-file implementation of offline RL algorithms in JAX☆143Updated 4 months ago
- A clean implementation of MuZero and AlphaZero following the AlphaZero General framework. Train and Pit both algorithms against each othe…☆157Updated 4 years ago
- Modular framework for Reinforcement Learning in python☆172Updated 2 years ago
- An API conversion tool for popular external reinforcement learning environments☆162Updated 3 months ago
- Experimentation with Regularized Nash Dynamics on a GPU accelerated game☆47Updated 2 years ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆317Updated last week
- Goal-Conditioned Reinforcement Learning with JAX☆142Updated 3 weeks ago
- A project that provides help for using DeepMind's mctx on gym-style environments.☆58Updated 5 months ago
- ☆42Updated 2 years ago
- Datasets for data-driven deep reinforcement learning with Atari (wrapper for datasets released by Google)☆118Updated 7 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆233Updated last month
- ☆341Updated 2 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆141Updated last year
- PyTorch implementation of DreamerV2 model-based RL algorithm☆216Updated 2 years ago
- ☆47Updated 11 months ago
- ☆41Updated 9 months ago
- Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.☆131Updated 8 months ago
- Simplifying Model-based RL: Learning Representations, Latent-space Models and Policies with One Objective☆80Updated 2 years ago
- ☆76Updated last month
- Implementation of Trajectory Transformer with attention caching and batched beam search☆111Updated 2 years ago
- [ICML 2020] PyTorch Code for "One Policy to Control Them All: Shared Modular Policies for Agent-Agnostic Control"☆228Updated 2 years ago
- MR.Q is a general-purpose model-free reinforcement learning algorithm.☆89Updated 2 weeks ago