code: def _make_elementary_quat(axis: int, angle: jax.Array) -> jax.Array:
      quat = jnp.zeros(4, dtype=angle.dtype)
      quat = quat.at[3].set(jnp.cos(angle / 2.0))
      quat = quat.at[axis].set(jnp.sin(angle / 2.0))
      return quat
  
imports: from __future__ import annotations
  import functools
  import re
  import typing
  import scipy.spatial.transform
  import jax
  import jax.numpy
  from jax._src.numpy.util import _wraps
label_desc: Use image node_id 14 for calling a global function with example usage: _make_elementary_quat(axis, angle) and returns: quat