* Fix ZeRO parameter initialization for tensors with `requires_grad=True`

* Simplify detach logic

---------
Co-authored-by: NLogan Adams <114770087+loadams@users.noreply.github.com>