I can easily move the CUDA initialization to `utils.py`, which I assume would be the natural home. Unless we want to make its own module -- depends on if you think of other places to easily use it.