I have created a script which will allow the new user to learn more about FSDP through hands on. I have tested the code on 4 A100 40GB each gpus. I have added support checkpointing as well for FSDP. I ...