Rockmate: an Efficient, Fast, Automatic and Generic Tool for Re-materialization in PyTorch

3 July 2023

Papers citing "Rockmate: an Efficient, Fast, Automatic and Generic Tool for Re-materialization in PyTorch"

2 / 2 papers shown

Title
ProTrain: Efficient LLM Training via Memory-Aware Techniques Hanmei Yang Jin Zhou Yao Fu Xiaoqun Wang Ramine Roane Hui Guan Tongping Liu VLM 28 0 0 12 Jun 2024
Mimose: An Input-Aware Checkpointing Planner for Efficient Training on GPU Jian-He Liao Mingzhen Li Qingxiao Sun Jiwei Hao F. Yu ... Ye Tao Zicheng Zhang Hailong Yang Zhongzhi Luan D. Qian 18 4 0 06 Sep 2022