class_docstr: A `CheckpointWrapper` is used to enable activation/gradient checkpointing on modules
  that you wrap via the `.wrap_module()` method.
code: def wrap_module(self, module: nn.Module, **kwargs) -> nn.Module:
      raise NotImplementedError
  
file_path: allennlp/allennlp/nn/checkpoint/checkpoint_wrapper.py
imports: import functools
  from typing import Callable, TypeVar
  import weakref
  import torch
  import torch.nn
  from torch.utils.checkpoint import CheckpointFunction
  from allennlp.common.registrable import Registrable
label_desc: Use image node_id 1 for calling the CheckpointWrapper obj's underlying member method code with example usage: obj.wrap_module(module) without return types