jax.lax.polygamma# jax.lax.polygamma(m, x)[source]# Elementwise polygamma: \(\psi^{(m)}(x)\). Parameters: m (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – x (Union[Array, ndarray, bool_, number, bool, int, float, complex]) – Return type: Array