google-deepmind / rlax
☆1,280Updated this week
Alternatives and similar repositories for rlax:
Users that are interested in rlax are comparing it to the libraries listed below
- JAX-based neural network library☆2,945Updated last month
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆799Updated 5 months ago
- ☆812Updated last week
- Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research☆714Updated 2 weeks ago
- Library for Model Based RL☆979Updated 6 months ago
- RL Environments in JAX 🌍☆682Updated 6 months ago
- Really Fast End-to-End Jax RL Implementations☆800Updated 4 months ago
- Optax is a gradient processing and optimization library for JAX.☆1,775Updated this week
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆651Updated 2 years ago
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent☆1,516Updated 9 months ago
- High throughput synchronous and asynchronous reinforcement learning☆860Updated 3 weeks ago
- A Python interface for reinforcement learning environments☆355Updated 2 years ago
- A library of reinforcement learning components and agents☆3,570Updated 2 weeks ago
- A toolkit for reproducible reinforcement learning research.☆1,912Updated last year
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,118Updated 5 months ago
- ☆866Updated 7 months ago
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆463Updated 9 months ago
- A PyTorch Platform for Distributed RL☆746Updated 3 years ago
- Reinforcement Learning in PyTorch☆2,239Updated 4 years ago
- ☆319Updated last year
- JAX - A curated list of resources https://github.com/google/jax☆1,676Updated 6 months ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆879Updated last year
- Dream to Control: Learning Behaviors by Latent Imagination☆623Updated 4 years ago
- Tonic RL library☆405Updated 6 months ago
- A PyTorch library for building deep reinforcement learning agents.☆649Updated 10 months ago
- A collection of reference environments for offline reinforcement learning☆1,400Updated 2 months ago
- 🕹️ A diverse suite of scalable reinforcement learning environments in JAX☆680Updated last month
- Python library for Reinforcement Learning.☆849Updated last week
- 🦁 A research-friendly codebase for fast experimentation of multi-agent reinforcement learning in JAX☆764Updated this week
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆943Updated 4 months ago