- Research Article
- 10.1109/lca.2026.3676331
JBOC: Just a Bunch of CXL-enabled SSDs for Resource-Efficient LLM Checkpointing
- Jan 01, 2026
- IEEE Computer Architecture Letters
- Hakbeom Jang + 5 more +5
The rapid growth of Large Language Models (LLMs) has pushed training systems to rely on memory sharding and offloading, introducing substantial data movement overhead. This is further exacerbated by frequent checkpointing, where training states must traverse multiple software and hardware layers. Such a process creates a critical performance bottleneck, as traditional host-centric hierarchies struggle to handle the massive data movement required for high-frequency, faulttolerant checkpointing. In this paper, we introduce JBOC, an in-device processing architecture leveraging CXL-enabled SSDs that fuses optimizer updates and checkpointing into a single, unified operation. By offloading massive footprint optimizer updates and executing these fused tasks directly within internal computational units, JBOC achieves highly efficient checkpointing as persistence is handled during the update pipeline. This offloaded approach allows training states to remain within deviceside resources, bypassing the host-side resource hierarchy and effectively isolating the host from heavy storage traffic. We validate JBOC on commercial CXL platforms from Marvell and SK Hynix. Our evaluation achieves up to a <inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"><tex-math notation="LaTeX">$4.40\times$</tex-math></inline-formula> speedup under per-iteration checkpointing. Compared to ZeRO-Offload, JBOC delivers comparable throughput while checkpointing <inline-formula xmlns:mml="http://www.w3.org/1998/Math/MathML" xmlns:xlink="http://www.w3.org/1999/xlink"><tex-math notation="LaTeX">$10\times$</tex-math></inline-formula> more frequently. Furthermore, JBOC outperforms both ZeROOffload and ZeRO-Infinity in memory-constrained scenarios, demonstrating its robustness when host resources are limited.
Read more