Switch to tensorflow-probability for gpu. (#1362)
Switch to tensorflow-probability for gpu. (#1362)

Switch to tensorflow-probability for gpu since tensorflow-probability-gpu is deprecated.
jesseengel/magenta,adarob/magenta,magenta/magenta,jesseengel/magenta,adarob/magenta,magenta/magenta