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