Assumed Density Filtering and Smoothing with Neural Network Surrogate Models