tensorplay.distributed.checkpoint.async_save

tensorplay.distributed.checkpoint.async_save(state_dict, *, checkpoint_id=None, storage_writer=None, planner=None, process_group=None, async_checkpointer_type: AsyncCheckpointerType = AsyncCheckpointerType.THREAD, async_stager: AsyncStager | None = None, no_dist=False, use_collectives=True) → Future[Any] | AsyncSaveResponse[source]

Stage the input and execute the checkpoint write asynchronously.