rlpyt: A Research Code Base for Deep Reinforcement Learning in PyTorch
Not on the map. Why?
Abstract (arXiv)
Since the recent advent of deep reinforcement learning for game play and simulated robotic control, a multitude of new algorithms have flourished. Most are model-free algorithms which can be categorized into three families: deep Q-learning, policy gradients, and Q-value policy gradients. These have developed along separate lines of research, such that few, if any, code bases incorporate all three kinds. Yet these algorithms share a great depth of common deep reinforcement learning machinery. We are pleased to share rlpyt, which implements all three algorithm families on top of a shared, optimized infrastructure, in a single repository. It contains modular implementations of many common deep RL algorithms in Python using PyTorch, a leading deep learning library. rlpyt is designed as a high-throughput code base for small- to medium-scale research in deep RL. This white paper summarizes its features, algorithms implemented, and relation to prior work, and concludes with detailed implementation and usage notes. rlpyt is available at https://github.com/astooke/rlpyt.
Key facts. Cited by 38 papers in the map (data through 7 Oct 2026). First posted on arXiv 3 Sep 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2020: 2; 2021: 15; 2022: 9; 2023: 8; 2024: 3; 2025: 1.
38 papers.
Sort:
By area:
- 2021Habitat 2.0: Training Home Assistants to Rearrange their Habitatcited by 501
- 2021Tianshou: a Highly Modularized Deep Reinforcement Learning Librarycited by 139
- 2022CORL: Research-oriented Deep Offline Reinforcement Learning Librarycited by 117
- 2021Pretraining Representations for Data-Efficient Reinforcement Learningcited by 105
- 2023TorchRL: A data-driven decision-making library for PyTorchcited by 59
- 2023Read and Reap the Rewards: Learning to Play Atari with the Help of Instruction Manualscited by 45
- 2022skrl: Modular and Flexible Library for Reinforcement Learningcited by 35
- 2021Multi-Agent MDP Homomorphic Networkscited by 24
- 2021On Lottery Tickets and Minimal Task Representations in Deep Reinforcement Learningcited by 23
- 2021Successor Feature Landmarks for Long-Horizon Goal-Conditioned Reinforcement Learningcited by 23
- 2020Inverse Reinforcement Learning with Natural Language Goalscited by 20
- 2021Megaverse: Simulating Embodied Agents at One Million Experiences per Secondcited by 19
26 more
- 2022Compositional Multi-Object Reinforcement Learning with Linear Relation Networkscited by 19
- 2021Large Batch Simulation for Deep Reinforcement Learningcited by 18
- 2021Average-Reward Reinforcement Learning with Trust Region Methodscited by 18
- 2022Explain My Surprise: Learning Efficient Long-Term Memory by Predicting Uncertain Outcomescited by 11
- 2023SRL: Scaling Distributed Reinforcement Learning to Over Ten Thousand Corescited by 10
- 2020Regularized Inverse Reinforcement Learningcited by 9
- 2021ElegantRL-Podracer: Scalable and Elastic Library for Cloud-Native Deep Reinforcement Learningcited by 9
- 2022Consistent Dropout for Policy Gradient Reinforcement Learningcited by 9
- 2024Acceleration for Deep Reinforcement Learning using Parallel and Distributed Computing: A Surveycited by 8
- 2021Spatial Graph Attention and Curiosity-driven Policy for Antiviral Drug Discoverycited by 5
- 2021Dr Jekyll and Mr Hyde: the Strange Case of Off-Policy Policy Updatescited by 5
- 2022RobustAnalog: Fast Variation-Aware Analog Circuit Design Via Multi-task RLcited by 5
- 2022Towards Trustworthy Automatic Diagnosis Systems by Emulating Doctors' Reasoning with Deep Reinforcement Learningcited by 5
- 2023GEAR: A GPU-Centric Experience Replay System for Large Reinforcement Learning Modelscited by 5
- 2021Low-Precision Reinforcement Learning: Running Soft Actor-Critic in Half Precisioncited by 3
- 2021Cogment: Open Source Framework For Distributed Multi-actor Training, Deployment & Operationscited by 3
- 2023RLLTE: Long-Term Evolution Project of Reinforcement Learningcited by 3
- 2024Tilted Quantile Gradient Updates for Quantile-Constrained Reinforcement Learningcited by 2
- 2022Karolos: An Open-Source Reinforcement Learning Framework for Robot-Task Environmentscited by 1
- 2023Revisiting Estimation Bias in Policy Gradients for Deep Reinforcement Learningcited by 1
- 2023Intrinsic Rewards from Self-Organizing Feature Maps for Exploration in Reinforcement Learningcited by 1
- 2023Spreeze: High-Throughput Parallel Reinforcement Learning Frameworkcited by 1
- 2024Towards Fault Tolerance in Multi-Agent Reinforcement Learningcited by 1
- 2025Tempo: Compiled Dynamic Deep Learning with Symbolic Dependence Graphscited by 1
- 2021The AI Arena: A Framework for Distributed Multi-Agent Reinforcement Learningcited by 0
- 2022Using Forwards-Backwards Models to Approximate MDP Homomorphismscited by 0
- 2025Tempo: Compiled Dynamic Deep Learning with Symbolic Dependence Graphs9 Jan 2025
- 2024Tilted Quantile Gradient Updates for Quantile-Constrained Reinforcement Learning17 Dec 2024
- 2024Towards Fault Tolerance in Multi-Agent Reinforcement Learning30 Nov 2024
- 2024Acceleration for Deep Reinforcement Learning using Parallel and Distributed Computing: A Survey8 Nov 2024
- 2023Spreeze: High-Throughput Parallel Reinforcement Learning Framework11 Dec 2023
- 2023GEAR: A GPU-Centric Experience Replay System for Large Reinforcement Learning Models8 Oct 2023
- 2023RLLTE: Long-Term Evolution Project of Reinforcement Learning28 Sep 2023
- 2023SRL: Scaling Distributed Reinforcement Learning to Over Ten Thousand Cores29 Jun 2023
- 2023TorchRL: A data-driven decision-making library for PyTorch1 Jun 2023
- 2023Read and Reap the Rewards: Learning to Play Atari with the Help of Instruction Manuals9 Feb 2023
- 2023Intrinsic Rewards from Self-Organizing Feature Maps for Exploration in Reinforcement Learning6 Feb 2023
- 2023Revisiting Estimation Bias in Policy Gradients for Deep Reinforcement Learning20 Jan 2023
26 more
- 2022Karolos: An Open-Source Reinforcement Learning Framework for Robot-Task Environments1 Dec 2022
- 2022Towards Trustworthy Automatic Diagnosis Systems by Emulating Doctors' Reasoning with Deep Reinforcement Learning13 Oct 2022
- 2022CORL: Research-oriented Deep Offline Reinforcement Learning Library13 Oct 2022
- 2022Using Forwards-Backwards Models to Approximate MDP Homomorphisms14 Sep 2022
- 2022Explain My Surprise: Learning Efficient Long-Term Memory by Predicting Uncertain Outcomes27 Jul 2022
- 2022RobustAnalog: Fast Variation-Aware Analog Circuit Design Via Multi-task RL13 Jul 2022
- 2022Consistent Dropout for Policy Gradient Reinforcement Learning23 Feb 2022
- 2022skrl: Modular and Flexible Library for Reinforcement Learning8 Feb 2022
- 2022Compositional Multi-Object Reinforcement Learning with Linear Relation Networks31 Jan 2022
- 2021ElegantRL-Podracer: Scalable and Elastic Library for Cloud-Native Deep Reinforcement Learning11 Dec 2021
- 2021Successor Feature Landmarks for Long-Horizon Goal-Conditioned Reinforcement Learning18 Nov 2021
- 2021Multi-Agent MDP Homomorphic Networks9 Oct 2021
- 2021Dr Jekyll and Mr Hyde: the Strange Case of Off-Policy Policy Updates29 Sep 2021
- 2021Tianshou: a Highly Modularized Deep Reinforcement Learning Library29 Jul 2021
- 2021Megaverse: Simulating Embodied Agents at One Million Experiences per Second17 Jul 2021
- 2021Habitat 2.0: Training Home Assistants to Rearrange their Habitat28 Jun 2021
- 2021Cogment: Open Source Framework For Distributed Multi-actor Training, Deployment & Operations21 Jun 2021
- 2021Pretraining Representations for Data-Efficient Reinforcement Learning9 Jun 2021
- 2021Average-Reward Reinforcement Learning with Trust Region Methods7 Jun 2021
- 2021Spatial Graph Attention and Curiosity-driven Policy for Antiviral Drug Discovery4 Jun 2021
- 2021On Lottery Tickets and Minimal Task Representations in Deep Reinforcement Learning4 May 2021
- 2021Large Batch Simulation for Deep Reinforcement Learning12 Mar 2021
- 2021The AI Arena: A Framework for Distributed Multi-Agent Reinforcement Learning9 Mar 2021
- 2021Low-Precision Reinforcement Learning: Running Soft Actor-Critic in Half Precision26 Feb 2021
- 2020Regularized Inverse Reinforcement Learning7 Oct 2020
- 2020Inverse Reinforcement Learning with Natural Language Goals16 Aug 2020
- 2022CORL: Research-oriented Deep Offline Reinforcement Learning Librarycited by 117
- 2020Inverse Reinforcement Learning with Natural Language Goalscited by 20
- 2020Regularized Inverse Reinforcement Learningcited by 9
- 2022CORL: Research-oriented Deep Offline Reinforcement Learning Library13 Oct 2022
- 2020Regularized Inverse Reinforcement Learning7 Oct 2020
- 2020Inverse Reinforcement Learning with Natural Language Goals16 Aug 2020
- 2021Habitat 2.0: Training Home Assistants to Rearrange their Habitatcited by 501
- 2021Tianshou: a Highly Modularized Deep Reinforcement Learning Librarycited by 139
- 2021Pretraining Representations for Data-Efficient Reinforcement Learningcited by 105
- 2023TorchRL: A data-driven decision-making library for PyTorchcited by 59
- 2023Read and Reap the Rewards: Learning to Play Atari with the Help of Instruction Manualscited by 45
- 2022skrl: Modular and Flexible Library for Reinforcement Learningcited by 35
- 2021Multi-Agent MDP Homomorphic Networkscited by 24
- 2021On Lottery Tickets and Minimal Task Representations in Deep Reinforcement Learningcited by 23
- 2021Successor Feature Landmarks for Long-Horizon Goal-Conditioned Reinforcement Learningcited by 23
- 2021Megaverse: Simulating Embodied Agents at One Million Experiences per Secondcited by 19
- 2022Compositional Multi-Object Reinforcement Learning with Linear Relation Networkscited by 19
- 2021Large Batch Simulation for Deep Reinforcement Learningcited by 18
12 most cited of 35 All 35 (JSON)
- 2025Tempo: Compiled Dynamic Deep Learning with Symbolic Dependence Graphs9 Jan 2025
- 2024Tilted Quantile Gradient Updates for Quantile-Constrained Reinforcement Learning17 Dec 2024
- 2024Towards Fault Tolerance in Multi-Agent Reinforcement Learning30 Nov 2024
- 2024Acceleration for Deep Reinforcement Learning using Parallel and Distributed Computing: A Survey8 Nov 2024
- 2023Spreeze: High-Throughput Parallel Reinforcement Learning Framework11 Dec 2023
- 2023GEAR: A GPU-Centric Experience Replay System for Large Reinforcement Learning Models8 Oct 2023
- 2023RLLTE: Long-Term Evolution Project of Reinforcement Learning28 Sep 2023
- 2023SRL: Scaling Distributed Reinforcement Learning to Over Ten Thousand Cores29 Jun 2023
- 2023TorchRL: A data-driven decision-making library for PyTorch1 Jun 2023
- 2023Read and Reap the Rewards: Learning to Play Atari with the Help of Instruction Manuals9 Feb 2023
- 2023Intrinsic Rewards from Self-Organizing Feature Maps for Exploration in Reinforcement Learning6 Feb 2023
- 2023Revisiting Estimation Bias in Policy Gradients for Deep Reinforcement Learning20 Jan 2023
12 newest of 35 All 35 (JSON)