[FSDP][Easy] Move utils to `_utils.py` (#84212)
I pulled this out into a separate PR. This just moves some utility functions to `_utils.py`.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/84212
Approved by: https://github.com/rohan-varma