TorchRL：面向 PyTorch 的数据驱动决策库
机器学习
2023-11-28 v2 人工智能
摘要
PyTorch 已崛起为一流的机器学习框架，但其缺乏适用于处理复杂现实数据与环境的大型开发团队的、原生的综合性决策与控制任务库。为解决此问题，我们提出 TorchRL，一个面向 PyTorch 的通用控制库，提供集成良好但相互独立的组件。我们引入了一种新颖且灵活的 PyTorch 原语 TensorDict，以促进强化学习（RL）与控制诸多分支间的精简算法开发。我们提供了构建模块的详尽描述以及该库跨领域与任务的广泛概览。最后，我们通过实验证明其可靠性与灵活性，并展示对比基准以表明其计算效率。TorchRL 提供长期支持，并在 GitHub 上公开可用，以促进研究界的复现与协作。代码已在 GitHub 上开源。
引用
@article{arxiv.2306.00577,
title = {TorchRL: A data-driven decision-making library for PyTorch},
author = {Albert Bou and Matteo Bettini and Sebastian Dittert and Vikash Kumar and Shagun Sodhani and Xiaomeng Yang and Gianni De Fabritiis and Vincent Moens},
journal= {arXiv preprint arXiv:2306.00577},
year = {2023}
}