torchsnapshot (pypi): A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.
Keywords: pytorch|snapshot|checkpoint