s.
  Parameters:
     * **sharded_optim_state_dict** (*Dict**[**str**, **Any**]*)
       -- Optimizer state dict corresponding to the unflattened
       parameters and holding the sharded optimizer state.

     * **model** (*torch.nn.Module*) -- Refer to
       :meth:"shard_full_optim_state_dict".

     * **optim** (*torch.optim.Optimizer*) -- Optimizer for
       "model" 's

     * **parameters.** --

  Returns:
     Refer to "shard_full_optim_state_dict()".

  Return type:
     *Dict*[str, *Any*]

forward(args, *kwargs)
  Runs the forward pass for the wrapped module, inserting FSDP-
  specific pre- and post-forward sharding logic.

  Return type:
     *Any*

static fsdp_modules(module, root_only=False)
  Returns all nested FSDP instances, possibly including "module"
  itself and only including FSDP root modules if "root_only=True".

  Parameters:
