Runs fine on CPU.
Source: https://github.com/HyperGAN/HyperGAN/blob/master/examples/2d-distribution.py