google-deepmind / rlax
☆1,234Updated 3 months ago
Related projects: ⓘ
- [NeurIPS'21 Outstanding Paper] Library for reliable evaluation on RL and ML benchmarks, even with only a handful of seeds.☆744Updated last month
- Reverb is an efficient and easy-to-use data storage and transport system designed for machine learning research☆704Updated last month
- bsuite is a collection of carefully-designed experiments that investigate core capabilities of a reinforcement learning (RL) agent☆1,503Updated 5 months ago
- ☆759Updated 2 months ago
- Library for Model Based RL☆953Updated 2 months ago
- RL Environments in JAX 🌍☆607Updated 2 months ago
- Optax is a gradient processing and optimization library for JAX.☆1,640Updated this week
- JAX-based neural network library☆2,860Updated last week
- A PyTorch Platform for Distributed RL☆737Updated 3 years ago
- DQN Zoo is a collection of reference implementations of reinforcement learning agents developed at DeepMind based on the Deep Q-Network (…☆450Updated 5 months ago
- Really Fast End-to-End Jax RL Implementations☆671Updated last week
- High throughput synchronous and asynchronous reinforcement learning☆795Updated 2 weeks ago
- A Python interface for reinforcement learning environments☆344Updated last year
- JAX (Flax) implementation of algorithms for Deep Reinforcement Learning with continuous action spaces.☆607Updated last year
- Reinforcement Learning in PyTorch☆2,219Updated 3 years ago
- ☆828Updated 2 months ago
- Open-source codebase for EfficientZero, from "Mastering Atari Games with Limited Data" at NeurIPS 2021.☆860Updated 8 months ago
- A curated list of awesome Deep Reinforcement Learning resources.☆659Updated last month
- A Graph Neural Network Library in Jax☆1,357Updated 6 months ago
- Massively parallel rigidbody physics simulation on accelerator hardware.☆2,220Updated 2 months ago
- A library of reinforcement learning components and agents☆3,469Updated 3 weeks ago
- SEED RL: Scalable and Efficient Deep-RL with Accelerated Central Inference. Implements IMPALA and R2D2 algorithms in TF2 with SEED's arch…☆794Updated last year
- Tonic RL library☆401Updated last month
- C++-based high-performance parallel environment execution engine (vectorized env) for general RL environments.☆1,066Updated last month
- A PyTorch library for building deep reinforcement learning agents.☆642Updated 6 months ago
- A toolkit for reproducible reinforcement learning research.☆1,858Updated last year
- Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning Framework on a GPU (JMLR 2022)☆457Updated last month
- A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.☆2,208Updated this week
- ☆309Updated last year
- Dream to Control: Learning Behaviors by Latent Imagination☆599Updated 4 years ago