A fast heuristic to optimize time-space tradeoff for large models. - Akifumi Imanishi; Zijian Xu 004; Masayuki Takagi; Sixue Wang; Emilio Castillo (NeurIPS, 2023)

A Heuristic for Periodic Memory Allocation with Little Fragmentation to Train Neural Networks. - Akifumi Imanishi; Zijian Xu 004 (ISMM, 2024)