tensorplay.utils.checkpoint.create_selective_checkpoint_contexts

tensorplay.utils.checkpoint.create_selective_checkpoint_contexts(policy_fn_or_list: Callable[[...], Any] | list[Any], allow_cache_entry_mutation: bool = False)[source]