How to use GPU to [chain1,chian2]?

[Runfa-Zhang]

Using GPUs

using Lux
chain = Chain(Dense(3,inner,Lux.σ),
              Dense(inner,inner,Lux.σ),
              Dense(inner,inner,Lux.σ),
              Dense(inner,inner,Lux.σ),
              Dense(inner,1)) 
ps = Lux.setup(Random.default_rng(), chain)[1]
ps = ps |> Lux.ComponentArray |> gpu .|> Float64

when chain=[chain1,chain2,...], how to use GPU?

[ChrisRackauckas]

Just make ps be an array of gpu'd initialized values.