Error when running Code on GPU

Hi everybody!

I am still quite new to Julia, so I hope I’m not missing something obvious. My goal is to solve a system of chemical equations with NeuralPDE. This runs fine, although very slow, on the CPU, but when I try to run it on the GPU to speed up the training, I am running into problems. The working example of my code is in this file:
solution_Lux_working_example.jl (8.4 KB)
the error I get is in this file:
Error.jl (70.2 KB)
(it’s not the complete stacktrace, I had to shorten it a little bit, otherwise I couldn’t upload it…)

Can somebody explain to me what that means?
Thanks in advance!