FullyShardedDataParallel ======================== .. automodule:: torch.distributed.fsdp .. autoclass:: torch.distributed.fsdp.FullyShardedDataParallel :members: .. autoclass:: torch.distributed.fsdp.BackwardPrefetch :members: .. autoclass:: torch.distributed.fsdp.ShardingStrategy :members: .. autoclass:: torch.distributed.fsdp.MixedPrecision :members: .. autoclass:: torch.distributed.fsdp.CPUOffload :members: