WarpDrive: Extremely Fast End-to-End Deep Multi-Agent Reinforcement Learning on a GPU


GitHub – salesforce/warp-drive
WarpDrive is a flexible, lightweight, and easy-to-use open-source reinforcement learning (RL) framework that implements end-to-end multi-agent RL on a single GPU (Graphics Processing Unit). Using the extreme parallelization capability of GPUs, WarpDrive enables orders-of-magnitude faster RL compared …