Imported from https://github.com/dayihengliu/Mu-Forcing-VRAE