← Search

Taebum Kim

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