code: def _get_jvp(primals: list[Any], tangents: list[Any], **params: Any):
      ref_primal, *idx = primals
      assert isinstance(ref_primal.aval, AbstractRef)
      ref_tangent, *_ = tangents
      assert isinstance(ref_tangent.aval, AbstractRef)
      return (
          get_p.bind(ref_primal, *idx, **params),
          get_p.bind(ref_tangent, *idx, **params),
      )
  
imports: from __future__ import annotations
  from functools import partial
  from typing import Any, Union
  import numpy
  from jax._src import ad_util
  from jax._src import core
  from jax._src import pretty_printer
  from jax._src import tree_util
  from jax._src.interpreters import ad
  from jax._src.interpreters import batching
  from jax._src.interpreters import partial_eval
  from jax._src.interpreters import mlir
  from jax._src.lax import lax
  from jax._src.typing import Array
  from jax._src.state import indexing
  from jax._src.state.types import AbstractRef, RefView, ReadEffect, WriteEffect, AccumEffect
  from jax._src.util import safe_map, safe_zip
label_desc: Use image node_id 21 for calling a global function with example usage: _get_jvp(primals, tangents) and returns: get_p,  get_p