class_docstr: Callable that returns a Jacobian computed by automatic differentiation.
class_docstr_tok: ["Callable","that","returns","a","Jacobian","computed","by","automatic","differentiation","."]
code: def __init__(self, ode_fn_vec):
      self._ode_fn_vec = ode_fn_vec
  
imports: import tensorflow.compat.v2
  from tensorflow_probability.python.internal import dtype_util
  from tensorflow_probability.python.internal import prefer_static
  from tensorflow_probability.python.math import gradient
  from tensorflow.python.util import nest
label_desc: Use image node_id 1 to create a new _AutomaticJacobian object from inherited base classes: object with example: obj = _AutomaticJacobian(ode_fn_vec)