Note: Applies to both scalar NNs and kernel NN
Note: Applies to both scalar NNs and kernel NN