Is it possible to prompt for restarting the machine after installation using WiX?
Yes. See the documentation for the REBOOT property in Windows Installer.
REBOOT
i.e.: