You shouldn’t create CUDA tensors in your init on a specific device.
Currently you are pushing self._coeffs to the default GPU.
Try to remove the .cuda() op and run it again.
PS: You can add code using three backticks `.