Hey I'm having some issues running the base examples from the docs. The problem seems to be FastChain initialising some parameters as `:nothing`. MWE: ```julia using DiffEqFlux U = FastChain( FastDense(2,5), FastDense(2,5)) @show intitial_params(U) # initial_params(U) = Union{Nothing, Float32}... ```