Note:
 The "requires_grad" state of each Tensor in "sample_args" must
 match the state that's expected for the corresponding real input
 in the training loop.

Warning:
 This API is in beta and may change in future releases.

Warning:
 "sample_args" for each callable must contain only Tensors. Other
 types are not allowed.

Warning:
 Returned callables do not support higher order differentiation
 (e.g., double backward).

Warning:
 In any "Module" passed to "make_graphed_callables()", only
 parameters may be trainable. Buffers must have
 "requires_grad=False".

Warning:
 After you pass a "torch.nn.Module" through
 "make_graphed_callables()", you may not add or remove any of that
 Module's parameters or buffers.

Warning:
 "torch.nn.Module"s passed to "make_graphed_callables()" must not
 have module hooks registered on them at the time they are passed.
 However, registering hooks on modules *after* passing them
