Add limit_all_gathers option to fsdp_config and fix forward_prefetch bug
fixes #21156 
Add limit_all_gathers option to fsdp_config.
Fix forward_prefetch bug.
_The documentation is not available anymore as the PR was closed or merged._<|||||>@sgugger  All changes resolved.