jax.lax.population_count# jax.lax.population_count(x)[source]# Elementwise popcount, count the number of set bits in each element. Parameters: x (Union[Array, ndarray, bool_, number, bool, int, float, complex]) Return type: Array