Type instability for simple MLP call using ComponentArrays

[ChrisRackauckas]

The tends to be a byproduct of Zygote. We are moving more and more to Enzyme which tends to be better with inference.