Add auto_wrap option in fairscale integration
# What does this PR do?

This PR adds support for the `auto_wrap` function added by fairscale to automatically wrap the model's modules in the `FSDP` container (necessary for ZeRO-DP3).

cc @stas00 So you are informed for when you want to experiment more with fairscale.