google-deepmind / rlaxLinks
☆1,362Updated 2 weeks ago
Alternatives and similar repositories for rlax
Users that are interested in rlax are comparing it to the libraries listed below
Sorting:
- RL Environments in JAX 🌍☆813Updated 4 months ago
- Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research☆746Updated 2 weeks ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆849Updated last year
- ☆884Updated last month
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent☆1,526Updated last year
- Library for Model Based RL☆1,024Updated last year
- ☆917Updated last year
- JAX-based neural network library☆3,103Updated 2 weeks ago
- High throughput synchronous and asynchronous reinforcement learning☆945Updated 4 months ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆718Updated 2 years ago
- Really Fast End-to-End Jax RL Implementations☆957Updated last year
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆763Updated this week
- Optax is a gradient processing and optimization library for JAX.☆2,039Updated last week
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆479Updated last year
- A Python interface for reinforcement learning environments☆377Updated 2 years ago
- Python library for Reinforcement Learning.☆910Updated 2 months ago
- SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference. Implements IMPALA and R2D2 algorithms in TF2 with SEED's arch…☆834Updated 2 years ago
- A PyTorch Platform for Distributed RL☆749Updated 4 years ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆912Updated last year
- Tonic RL library☆417Updated last year
- Evolution Strategies in JAX 🦎☆668Updated 3 weeks ago
- ♟️ Vectorized RL game environments in JAX☆532Updated 7 months ago
- ☆326Updated 2 years ago
- Massively parallel rigidbody physics simulation on accelerator hardware.☆2,882Updated last week
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆507Updated 2 weeks ago
- Dream to Control: Learning Behaviors by Latent Imagination☆665Updated 5 years ago
- Modular Deep Reinforcement Learning framework in PyTorch. Companion library of the book "Foundations of Deep Reinforcement Learning".☆1,304Updated last week
- A library of reinforcement learning components and agents☆3,801Updated 2 weeks ago
- A toolkit for reproducible reinforcement learning research.☆2,030Updated 2 years ago
- ☆597Updated this week