coax-dev / coax
Modular framework for Reinforcement Learning in python
☆172Updated 2 years ago
Alternatives and similar repositories for coax
Users that are interested in coax are comparing it to the libraries listed below
Sorting:
- An API conversion tool for popular external reinforcement learning environments☆165Updated 4 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆440Updated 2 weeks ago
- JAX implementations of core Deep RL algorithms☆79Updated 3 years ago
- A collection of RL algorithms written in JAX.☆97Updated 2 years ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆235Updated last month
- Recurrent and multi-process PyTorch implementation of deep reinforcement Actor-Critic algorithms A2C and PPO☆200Updated 2 years ago
- Baselines for gymnax 🤖☆66Updated 2 years ago
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆205Updated last month
- Benchmarking RL generalization in an interpretable way.☆156Updated 2 months ago
- ☆299Updated 4 months ago
- RL Environments in JAX 🌍☆744Updated 10 months ago
- A Python interface for reinforcement learning environments☆364Updated 2 years ago
- Accelerated minigrid environments with JAX☆135Updated 9 months ago
- Vectorized interface for reinforcement learning environments☆140Updated 2 years ago
- Reinforcement learning library in JAX.☆100Updated last year
- Real-World RL Benchmark Suite☆354Updated 4 years ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆678Updated 2 years ago
- ☆101Updated last year
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆322Updated last week
- ☆228Updated 5 months ago
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆302Updated last year
- Accelerated Quality-Diversity☆294Updated last week
- Keeping track of RL experiments☆161Updated 2 years ago
- OpenAI Gym wrapper for the DeepMind Control Suite☆215Updated 11 months ago
- An easy-to-use reinforcement learning library for research and education.☆167Updated last week
- Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.☆130Updated 8 months ago
- A clean implementation of MuZero and AlphaZero following the AlphaZero General framework. Train and Pit both algorithms against each othe…☆158Updated 4 years ago
- impact-driven-exploration☆131Updated last year
- From search engines, to science, to robotics, this reposity is meant to showcase the use of reinforcement learning in the world..☆234Updated 10 months ago
- PAIRED in PyTorch 🔥☆59Updated 2 years ago