google-deepmind / rlax
☆1,289Updated last month
Alternatives and similar repositories for rlax:
Users that are interested in rlax are comparing it to the libraries listed below
- Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research☆718Updated last month
- RL Environments in JAX 🌍☆709Updated 7 months ago
- ☆827Updated this week
- Library for Model Based RL☆986Updated 7 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆807Updated 6 months ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆661Updated 2 years ago
- A PyTorch Platform for Distributed RL☆746Updated 3 years ago
- Optax is a gradient processing and optimization library for JAX.☆1,809Updated this week
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆465Updated 10 months ago
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent☆1,518Updated 10 months ago
- JAX-based neural network library☆2,973Updated last month
- Really Fast End-to-End Jax RL Implementations☆823Updated 5 months ago
- High throughput synchronous and asynchronous reinforcement learning☆872Updated 2 months ago
- A Python interface for reinforcement learning environments☆356Updated 2 years ago
- ☆882Updated 8 months ago
- Python library for Reinforcement Learning.☆859Updated 2 weeks ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆694Updated 3 weeks ago
- Tonic RL library☆406Updated 7 months ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆884Updated last year
- A library of reinforcement learning components and agents☆3,597Updated last month
- SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference. Implements IMPALA and R2D2 algorithms in TF2 with SEED's arch…☆801Updated 2 years ago
- Code for Go-Explore: a New Approach for Hard-Exploration Problems☆561Updated 2 years ago
- Reinforcement Learning in PyTorch☆2,241Updated 4 years ago
- Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)☆477Updated 7 months ago
- Dream to Control: Learning Behaviors by Latent Imagination☆628Updated 4 years ago
- ☆319Updated last year
- ☆548Updated this week
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,129Updated 6 months ago
- PFRL: a PyTorch-based deep reinforcement learning library☆1,221Updated 6 months ago
- Evolution Strategies in JAX 🦎☆560Updated this week