Scalar Indexing Error in PINN Implementation on GPU

Thanks for your response.

I changed

to
ps = ps .|> ComponentArray |> gpu_dev.

It works.