lines 6-16 of file: python/cmpad/jax/gradient.py

{xrst_begin jax_gradient.py}

{xrst_template ,
   python/cmpad/gradient.xrst
   @Package@    , JAX
   @#######@    , ###
   @package@    , jax
   @algo_type@  , ``jax.numpy.array`` with ``float`` elements
}

{xrst_end jax_gradient.py}