I have implemented an extended kalman filter optimizer in tensorflow. I have to calculate the jacobian instead of gradient against the prediction generated by the model. The mod