google-deepmind / rlax
β1,303Updated 2 months ago
Alternatives and similar repositories for rlax:
Users that are interested in rlax are comparing it to the libraries listed below
- β841Updated 2 weeks ago
- Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning researchβ720Updated 3 weeks ago
- RL Environments in JAX πβ730Updated 8 months ago
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.β817Updated 7 months ago
- Optax is a gradient processing and optimization library for JAX.β1,847Updated this week
- JAX-based neural network libraryβ2,999Updated 2 weeks ago
- Library for Model Based RLβ989Updated 8 months ago
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.β670Updated 2 years ago
- Really Fast End-to-End Jax RL Implementationsβ845Updated 6 months ago
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agentβ1,518Updated 11 months ago
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (β¦β470Updated 11 months ago
- A Python interface for reinforcement learning environmentsβ362Updated 2 years ago
- A PyTorch Platform for Distributed RLβ746Updated 3 years ago
- High throughput synchronous and asynchronous reinforcement learningβ886Updated 3 months ago
- JAX - A curated list of resources https://github.com/google/jaxβ1,747Updated last month
- Python library for Reinforcement Learning.β861Updated this week
- β559Updated this week
- a Lightweight library for sequential learning agents, including reinforcement learningβ429Updated 2 years ago
- Tonic RL libraryβ406Updated 8 months ago
- Evolution Strategies in JAX π¦β587Updated last week
- πΉοΈ A diverse suite of scalable reinforcement learning environments in JAXβ712Updated last week
- RLMeta is a light-weight flexible framework for Distributed Reinforcement Learning Research.β282Updated 2 years ago
- Hardware accelerated, batchable and differentiable optimizers in JAX.β956Updated 2 weeks ago
- Massively parallel rigidbody physics simulation on accelerator hardware.β2,597Updated last month
- SBX: Stable Baselines Jax (SB3 + Jax) RL algorithmsβ425Updated last week
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.β886Updated last year
- SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference. Implements IMPALA and R2D2 algorithms in TF2 with SEED's archβ¦β806Updated 2 years ago
- Multi-Agent Reinforcement Learning with JAXβ549Updated 3 weeks ago
- Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)β480Updated 8 months ago
- Real-World RL Benchmark Suiteβ352Updated 4 years ago