[1/N] Introduce init_device_mesh() (#107254)
This PR introduces init_device_mesh() as an API to standardize UX device_mesh initialization.
The functionality of slicing out a submesh from a given mesh would come in later PRs.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/107254
Approved by: https://github.com/wanchaol