class_docstr: Adjoint solver which propagates step size between solves.
class_docstr_tok: ["Adjoint","solver","which","propagates","step","size","between","solves","."]
code: def _adjust_solver_internal_state_for_state_jump(self, **kwargs):
      return kwargs["previous_solver_internal_state"]
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.math import gradient
  from tensorflow_probability.python.math.ode import base
  from tensorflow_probability.python.math.ode import bdf
  from tensorflow_probability.python.math.ode import dormand_prince
label_desc: Use image node_id 4 for calling the StepSizeHeuristicAdjointSolver obj's underlying member method code with example usage: obj._adjust_solver_internal_state_for_state_jump() and returns: kwargs