coax-dev / coax
Modular framework for Reinforcement Learning in python
☆170Updated last year
Alternatives and similar repositories for coax:
Users that are interested in coax are comparing it to the libraries listed below
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆220Updated last week
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆381Updated this week
- Baselines for gymnax 🤖☆61Updated last year
- A collection of RL algorithms written in JAX.☆95Updated 2 years ago
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆230Updated last week
- Accelerated minigrid environments with JAX☆128Updated 5 months ago
- ☆178Updated last month
- An API conversion tool for popular external reinforcement learning environments☆148Updated 3 weeks ago
- Accelerated Quality-Diversity☆274Updated 3 months ago
- Recurrent and multi-process PyTorch implementation of deep reinforcement Actor-Critic algorithms A2C and PPO☆193Updated 2 years ago
- Benchmarking RL generalization in an interpretable way.☆138Updated 11 months ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆267Updated 3 weeks ago
- ☆100Updated 11 months ago
- ☆293Updated last month
- ☆211Updated 2 months ago
- RL Environments in JAX 🌍☆682Updated 6 months ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆651Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆293Updated last year
- Reinforcement learning library in JAX.☆102Updated last year
- Vectorized interface for reinforcement learning environments☆140Updated last year
- Partially Observable Process Gym☆174Updated 6 months ago
- Simple single-file baselines for Q-Learning in pure-GPU setting☆135Updated 2 months ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆335Updated 3 weeks ago
- From search engines, to science, to robotics, this reposity is meant to showcase the use of reinforcement learning in the world..☆220Updated 7 months ago
- ☆72Updated 2 months ago
- OpenAI Gym wrapper for the DeepMind Control Suite☆210Updated 8 months ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆109Updated 5 months ago
- Real-World RL Benchmark Suite☆349Updated 4 years ago
- Evaluating long-term memory of reinforcement learning algorithms☆138Updated last year