Remove DDP import (#89982)
This import is only used for typing, removing it to avoid circular ref
in next diffs
Differential Revision: [D41636897](https://our.internmc.facebook.com/intern/diff/D41636897/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/89982
Approved by: https://github.com/zhaojuanmao