← Search

Gyeong-In Yu

3 accepted papers

2023

BPipe: Memory-Balanced Pipeline Parallelism for Training Large Language Models

ICML 2023oral

Pipeline parallelism is a key technique for training large language models within GPU clusters. However, it often leads to a memory imbalance problem, where certain GPUs face high memory pressure while others underutilize their capacity. This imbalance results in suboptimal training performance, eve…

Cited by 34SourcePDFScholar
2021

Terra: Imperative-Symbolic Co-Execution of Imperative Deep Learning Programs

NeurIPS 2021poster

Imperative programming allows users to implement their deep neural networks (DNNs) easily and has become an essential part of recent deep learning (DL) frameworks. Recently, several systems have been proposed to combine the usability of imperative programming with the optimized performance of symbol…

Cited by 7SourcePDFScholar
2020

Nimble: Lightweight and Parallel GPU Task Scheduling for Deep Learning

NeurIPS 2020spotlight

Deep learning (DL) frameworks take advantage of GPUs to improve the speed of DL inference and training. Ideally, DL frameworks should be able to fully utilize the computation power of GPUs such that the running time depends on the amount of computation assigned to GPUs. Yet, we observe that in sched…