google-deepmind / rlaxLinks
β1,398Updated 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β761Updated this week
- RL Environments in JAX πβ847Updated 7 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.β865Updated last year
- β929Updated last year
- Library for Model Based RLβ1,042Updated last year
- β918Updated last week
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agentβ1,530Updated last year
- πΉοΈ A diverse suite of scalable reinforcement learning environments in JAXβ797Updated last month
- Really Fast End-to-End Jax RL Implementationsβ1,010Updated last year
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (β¦β491Updated last year
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.β741Updated 3 years ago
- A Python interface for reinforcement learning environmentsβ388Updated 3 years ago
- JAX-based neural network libraryβ3,164Updated last week
- High throughput synchronous and asynchronous reinforcement learningβ966Updated 2 months ago
- Optax is a gradient processing and optimization library for JAX.β2,152Updated this week
- A PyTorch Platform for Distributed RLβ752Updated 4 years ago
- Dream to Control: Learning Behaviors by Latent Imaginationβ700Updated 5 years ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.β921Updated 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β¦β838Updated 3 years ago
- Tonic RL libraryβ419Updated last year
- Evolution Strategies in JAX π¦β718Updated 3 months ago
- β329Updated 2 years ago
- βοΈ Vectorized RL game environments in JAXβ575Updated 10 months ago
- Code for Go-Explore: a New Approach for Hard-Exploration Problemsβ579Updated 3 years ago
- Python library for Reinforcement Learning.β917Updated 5 months ago
- Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)β499Updated 8 months ago
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsβ549Updated 3 weeks ago
- β614Updated last month
- A toolkit for reproducible reinforcement learning research.β2,069Updated 2 years ago
- RLMeta is a light-weight flexible framework for Distributed Reinforcement Learning Research.β283Updated 2 years ago