🚀 Feature

After pytorch/pytorch#78168 was apparently resolved on pytorch master/nightlies we should re-evaluate fp16/amp support for mps devices.

cc @Borda @carmocca @justusschock @awaelchli @akihironitta @rohitgr7