google-deepmind / rlaxLinks
☆1,373Updated last month
Alternatives and similar repositories for rlax
Users that are interested in rlax are comparing it to the libraries listed below
Sorting:
- Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research☆751Updated this week
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆852Updated last year
- ☆890Updated last week
- RL Environments in JAX 🌍☆828Updated 5 months ago
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent☆1,527Updated last year
- High throughput synchronous and asynchronous reinforcement learning☆952Updated this week
- Library for Model Based RL☆1,030Updated last year
- JAX-based neural network library☆3,117Updated last month
- ☆921Updated last year
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆728Updated 3 years ago
- Really Fast End-to-End Jax RL Implementations☆973Updated last year
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆484Updated last year
- A PyTorch Platform for Distributed RL☆749Updated 4 years ago
- Optax is a gradient processing and optimization library for JAX.☆2,082Updated this week
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆774Updated last week
- Tonic RL library☆418Updated last year
- A Python interface for reinforcement learning environments☆385Updated 2 years 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
- ☆329Updated 2 years ago
- Dream to Control: Learning Behaviors by Latent Imagination☆681Updated 5 years ago
- Python library for Reinforcement Learning.☆915Updated 3 months ago
- Evolution Strategies in JAX 🦎☆683Updated last month
- a Lightweight library for sequential learning agents, including reinforcement learning☆431Updated 2 years ago
- Code for Go-Explore: a New Approach for Hard-Exploration Problems☆578Updated 2 years ago
- RLMeta is a light-weight flexible framework for Distributed Reinforcement Learning Research.☆283Updated 2 years ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆915Updated last year
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithms☆523Updated last month
- Massively parallel rigidbody physics simulation on accelerator hardware.☆2,932Updated last week
- A PyTorch library for building deep reinforcement learning agents.☆652Updated last year
- Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)☆492Updated 6 months ago