google-deepmind / rlax
☆1,309Updated last week
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☆724Updated last month
- RL Environments in JAX 🌍☆742Updated 9 months ago
- Library for Model Based RL☆994Updated 9 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆825Updated 8 months ago
- JAX-based neural network library☆3,013Updated this week
- ☆845Updated 3 weeks ago
- A PyTorch Platform for Distributed RL☆747Updated 3 years ago
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆472Updated last year
- Optax is a gradient processing and optimization library for JAX.☆1,868Updated this week
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆674Updated 2 years ago
- A Python interface for reinforcement learning environments☆362Updated 2 years ago
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent☆1,518Updated last year
- Really Fast End-to-End Jax RL Implementations☆859Updated 7 months ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆893Updated last year
- SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference. Implements IMPALA and R2D2 algorithms in TF2 with SEED's arch…☆819Updated 2 years ago
- ☆897Updated 9 months ago
- Dream to Control: Learning Behaviors by Latent Imagination☆636Updated 4 years ago
- Reinforcement Learning in PyTorch☆2,250Updated 4 years ago
- A toolkit for reproducible reinforcement learning research.☆1,964Updated last year
- A PyTorch library for building deep reinforcement learning agents.☆652Updated last year
- RLMeta is a light-weight flexible framework for Distributed Reinforcement Learning Research.☆282Updated 2 years ago
- Evolution Strategies in JAX 🦎☆602Updated 3 weeks ago
- Code for Go-Explore: a New Approach for Hard-Exploration Problems☆564Updated 2 years ago
- Tonic RL library☆406Updated 9 months ago
- Python library for Reinforcement Learning.☆868Updated last week
- ☆565Updated last week
- A Graph Neural Network Library in Jax☆1,425Updated last year
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,147Updated 8 months ago
- MiniHack the Planet: A Sandbox for Open-Ended Reinforcement Learning Research☆494Updated 2 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆430Updated 3 weeks ago