That's interesting, I recently wrote a game of life that uses PyTorch to run on the GPU:

https://gist.github.com/olooney/d835353176bd9a679ad2f057cd58...

It benefits from using PyTorch's 2D convolution operation but sufferers it has to keep casting types to get it to work. Still, it's quite fast for very large grids!