pep8, new compute_residual function
tf-mxne demo