Provide an auto wrap policy for common transformer models
Pull Request resolved: https://github.com/pytorch/pytorch/pull/76455
Provide an auto wrap policy for common transformer models
Differential Revision: [D35972488](https://our.internmc.facebook.com/intern/diff/D35972488/)
Approved by: https://github.com/rohan-varma