they are nested under this instance. This argument may be used
        to avoid sharding specific parameters at module granularity
        when using an "auto_wrap_policy" or if parameters' sharding is
        not managed by FSDP. (Default: "None")
  * **param_init_fn**
    (*Optional**[**Callable**[**[**nn.Module**]**, **None**]**]*)
    --

    A "Callable[torch.nn.Module] -> None" that specifies how
    modules that are currently on the meta device should be
    initialized onto an actual device. Note that as of v1.12, we
    detect modules on the meta device via "is_meta" check and
    apply a default initialization that calls "reset_parameters"
    method on the passed in "nn.Module" if "param_init_fn" is not
    specified, otherwise we run "param_init_fn" to initialize the
    passed in "nn.Module". In particular, this means that if
    "is_meta=True" for any module parameters for modules that will
