84 citations · 518 across the 35 of their papers we have counts for
Showing cs.DCShow all
3 papers · 1 filter
cs.DC2022★ 7 cited
MGG: Accelerating Graph Neural Networks with Fine-grained intra-kernel Communication-Computation Pipelining on Multi-GPU Platforms
Yuke Wang, Boyuan Feng, Zheng Wang +4
The increasing size of input graphs for graph neural networks (GNNs) highlights the demand for using multi-GPU platforms. However, existing multi-GPU GNN systems optimize the compu…
cs.DC2022
GMI-DRL: Empowering Multi-GPU Deep Reinforcement Learning with GPU Spatial Multiplexing
Yuke Wang, Boyuan Feng, Zheng Wang +3
With the increasing popularity of robotics in industrial control and autonomous driving, deep reinforcement learning (DRL) raises the attention of various fields. However, DRL comp…
cs.DC2020★ 6 cited
MARS: Malleable Actor-Critic Reinforcement Learning Scheduler
Betis Baheri, Jacob Tronge, Bo Fang +3
In this paper, we introduce MARS, a new scheduling system for HPC-cloud infrastructures based on a cost-aware, flexible reinforcement learning approach, which serves as an intermed…