TransformerEngine : Add test with FSDP (and updates to ddp_wrapper
in test_ddp.py)
#118
This job succeeded
Loading
ddp_wrapper
in test_ddp.py)
#118