coax-dev / coaxLinks
Modular framework for Reinforcement Learning in python
☆175Updated 2 years ago
Alternatives and similar repositories for coax
Users that are interested in coax are comparing it to the libraries listed below
Sorting:
- JAX implementations of core Deep RL algorithms☆82Updated 3 years ago
- Baselines for gymnax 🤖☆71Updated 2 years ago
- A collection of RL algorithms written in JAX.☆104Updated 3 years ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆250Updated last month
- Accelerated Quality-Diversity☆318Updated 2 months ago
- Accelerated minigrid environments with JAX☆146Updated last week
- A bare-bones Python library for quality diversity optimization.☆238Updated last week
- Hardware-Accelerated Reinforcement Learning Algorithms in pure Jax!☆235Updated 3 months ago
- ☆101Updated last year
- Reinforcement learning library in JAX.☆100Updated last year
- Fully differentiable RL environments, written in Ivy.☆65Updated 2 years ago
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆348Updated this week
- RL Environments in JAX 🌍☆801Updated 3 months ago
- A Python interface for reinforcement learning environments☆377Updated 2 years ago
- An API conversion tool for popular external reinforcement learning environments☆181Updated last week
- Simple single-file baselines for Q-Learning in pure-GPU setting☆182Updated 5 months ago
- Real-World RL Benchmark Suite☆354Updated 5 years ago
- ☆83Updated this week
- DeepMind Alchemy task environment: a meta-reinforcement learning benchmark☆200Updated 2 years ago
- An easy-to-use reinforcement learning library for research and education.☆171Updated last week
- ☆311Updated 8 months ago
- CleanRL's implementation of DeepMind's Podracer Sebulba Architecture for Distributed DRL☆114Updated last year
- Jax implementation of Proximal Policy Optimization (PPO) specifically tuned for Procgen, with benchmarked results and saved model weights…☆58Updated 3 years ago
- ☆52Updated 2 years ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆500Updated 2 weeks ago
- Accelerated replay buffers in JAX☆43Updated 2 years ago
- Contains JAX implementation of algorithms for inverse reinforcement learning☆72Updated last year
- Benchmarking RL generalization in an interpretable way.☆161Updated 2 months ago
- megastep helps you build 1-million FPS reinforcement learning environments on a single GPU☆141Updated 3 years ago
- A grid-world game engine for game AI research☆245Updated last year