jaxpolylog.jax_polylog_vmap

jaxpolylog.jax_polylog_vmap#

jax_polylog_vmap(z: complex, s: int, p_range: int, approx: str = 'inf', pval: float = 0.23231338255518164) → complex[source]#

Description: Vectorized version of the polylogarithm function using JAX’s vmap.

Parameters:

  • z (complex) – The input values at which to evaluate the polylogarithm. Must be a 1D array.

  • s (int) – The order of the polylogarithm. Must be an integer.

  • p_range (int) – The number of terms to include in the series expansion for non-predefined s values. Higher values increase accuracy but also computation time.

  • approx (str, optional) – The approximation method to use. Must be one of “inf”, “integral”, “zero”, or “patch”. Default is "inf".

  • pval (float, optional) – Transition parameter for the "patch" method. See jax_polylog(). Default is _PVAL_OPTIMAL.

Returns:

complex – The computed polylogarithm values at the input z.