Q: Trying to run Mandelbrot on GPU via PyTorch, but not seeing speed-up