CUDA compatibility

[viking-sudo-rm]

I'm not quite sure if the StackNN code is fully CUDA compatible, but recently I had some issues trying to use it.

Should investigate this and add cuda options everywhere in structs/control_layer.

[viking-sudo-rm]

https://pytorch.org/docs/stable/notes/cuda.html