TransformerEngine : Add test with FSDP (and updates to ddp_wrapper
in test_ddp.py)
#594
Loading
ddp_wrapper
in test_ddp.py)
#594