With [`ParameterHandling`](https://github.com/JuliaGaussianProcesses/ParameterHandling.jl) we can simplify `build_nn_function` and also make it more general (to be used with GHNNs for example).