It looks like something goes wrong when trying to generate a component array of the NN parameters (it looks like somewhere the assumption was made that all layers have the same amounts of weights and biases, which is not the case). It could be a bug in how NeuralPDE.jl uses ComponentArrays.jl or in ComponentArrays.jl itself. Are your packages up to date?