Let's run [this planar flow test](https://github.com/TuringLang/NormalizingFlows.jl/blob/4a018d8734901e2cd05396a0e403ac802c2ec656/test/interface.jl#L65-L111) with GPU.
