RoboEden / jux
JUX is a JAX-accelerated engine for Lux AI Challenge 2022.
☆19Updated last year
Alternatives and similar repositories for jux:
Users that are interested in jux are comparing it to the libraries listed below
- ☆104Updated 3 years ago
- Baselines for gymnax 🤖☆63Updated last year
- A collection of RL algorithms written in JAX.☆95Updated 2 years ago
- An API conversion tool for popular external reinforcement learning environments☆151Updated last month
- Deep Hierarchical Planning from Pixels☆94Updated 2 years ago
- ☆214Updated 3 months ago
- TeachMyAgent is a testbed platform for Automatic Curriculum Learning methods in Deep RL.☆68Updated last year
- ☆48Updated 2 years ago
- Partially Observable Process Gym☆178Updated 7 months ago
- impact-driven-exploration☆130Updated last year
- A tool to automate installing Atari ROMs for the Arcade Learning Environment☆78Updated last year
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆225Updated last month
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆109Updated 6 months ago
- Datasets for data-driven deep reinforcement learning with Atari (wrapper for datasets released by Google)☆113Updated 5 months ago
- Emergent collective intelligence from massive-agent cooperation and competition☆25Updated 2 years ago
- Implementation of Trajectory Transformer with attention caching and batched beam search☆109Updated last year
- Simple single-file baselines for Q-Learning in pure-GPU setting☆137Updated 2 months ago
- ☆41Updated last year
- Highly scalable 2D JAX physics engine.☆45Updated 3 weeks ago
- OpenAI Gym wrapper for the DeepMind Control Suite☆210Updated 9 months ago
- Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.☆128Updated 6 months ago
- Vectorized interface for reinforcement learning environments☆139Updated last year
- ☆248Updated 3 years ago
- Clean single-file implementation of offline RL algorithms in JAX☆132Updated last month
- official implementation for our paper Cal-QL: Calibrated Offline RL Pre-Training for Efficient Online Fine-Tuning☆84Updated 6 months ago
- A project that provides help for using DeepMind's mctx on gym-style environments.☆54Updated 3 months ago
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- General Modules for JAX☆63Updated 6 months ago
- Revisiting Rainbow☆74Updated 3 years ago
- Benchmarking RL generalization in an interpretable way.☆142Updated last week