Is your feature request related to a problem? Please describe.
Although its is more of a naming issue, but I think we should move to lightning as a core dep in place of pytorch_lightning as Lightning AI has moved to the new naming. For now, both the packages are released simultaneously, but we can't say when this may stop. So, we should fully migrate to lightning.pytorch.