jax.scipy.special.zeta

Contents

jax.scipy.special.zeta#

jax.scipy.special.zeta = <jax._src.custom_derivatives.custom_jvp object>[source]#

The Hurwitz zeta function.

JAX implementation of scipy.special.zeta(). JAX does not implement the Riemann zeta function (i.e. q = None).

\[\zeta(x, q) = \sum_{n=0}^\infty \frac{1}{(n + q)^x}\]
Parameters:
  • x (ArrayLike) – arraylike, real-valued

  • q (ArrayLike | None) – arraylike, real-valued

Returns:

array of zeta function values

Return type:

Array