tensorplay.distributed.fsdp.fully_shard

tensorplay.distributed.fsdp.fully_shard(module: Any, *, mesh: Any = None, reshard_after_forward: bool | int | None = None, shard_placement_fn: Callable[[Any], Any] | None = None, mp_policy: MixedPrecisionPolicy | None = None, offload_policy: OffloadPolicy | None = None, ignored_params: set[Any] | None = None, dp_mesh_dims: DataParallelMeshDims | None = None) → Any[source]