coax-dev / coax
Modular framework for Reinforcement Learning in python
☆165Updated last year
Related projects: ⓘ
- SBX: Stable Baselines Jax (SB3 + Jax)☆321Updated 3 weeks ago
- JAX implementations of core Deep RL algorithms☆79Updated 2 years ago
- An API conversion tool for popular external reinforcement learning environments☆131Updated 3 months ago
- ⚡ Flashbax: Accelerated Replay Buffers in JAX☆195Updated 3 weeks ago
- A collection of RL algorithms written in JAX.☆92Updated 2 years ago
- RL Environments in JAX 🌍☆607Updated 2 months ago
- ☆141Updated 2 weeks ago
- ☆282Updated last year
- Real-World RL Benchmark Suite☆343Updated 4 years ago
- Accelerated Quality-Diversity☆261Updated this week
- A Python interface for reinforcement learning environments☆344Updated last year
- 🏛️A research-friendly codebase for fast experimentation of single-agent reinforcement learning in JAX • End-to-End JAX RL☆204Updated this week
- Benchmarking RL generalization in an interpretable way.☆128Updated 7 months ago
- Recurrent and multi-process PyTorch implementation of deep reinforcement Actor-Critic algorithms A2C and PPO☆191Updated last year
- Implementations of robust Dual Curriculum Design (DCD) algorithms for unsupervised environment design.☆121Updated 3 weeks ago
- A standard format for offline reinforcement learning datasets, with popular reference datasets and related utilities☆260Updated this week
- ☆100Updated 7 months ago
- A bare-bones Python library for quality diversity optimization.☆207Updated last month
- Accelerated minigrid environments with JAX☆102Updated last month
- (Crafter + NetHack) in JAX. ICML 2024 Spotlight.☆189Updated 2 weeks ago
- Baselines for gymnax 🤖☆57Updated last year
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆608Updated last year
- megastep helps you build 1-million FPS reinforcement learning environments on a single GPU☆131Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imagination, implemented in PyTorch.☆277Updated 8 months ago
- From search engines, to science, to robotics, this reposity is meant to showcase the use of reinforcement learning in the world..☆214Updated 3 months ago
- PAIRED in PyTorch 🔥☆56Updated last year
- An easy-to-use reinforcement learning library for research and education.☆162Updated last month
- Partially Observable Process Gym☆158Updated 2 months ago
- Reinforcement learning library in JAX.☆102Updated 10 months ago
- ☆328Updated last year