fix (JAX backends)(statistical.py): adding a check to the implementat…

…ion for `sum` to cast the output to an integer in cases when the input is a boolean array. This is also inline with the behavior of `jnp.sum`

ivy/functional/backends/jax/statistical.py

@@ -1,4 +1,5 @@
 # global
+import jax
 import jax.numpy as jnp
 from typing import Union, Optional, Sequence
 
@@ -112,6 +113,11 @@ def sum(
     if dtype != x.dtype and not ivy.is_bool_dtype(x):
         x = jnp.astype(x, dtype)
     axis = tuple(axis) if isinstance(axis, list) else axis
+    if ivy.is_bool_dtype(x):
+        if jax.config.jax_enable_x64:
+            dtype = ivy.as_native_dtype('int64')
+        else:
+            dtype = ivy.as_native_dtype('int32')
     return jnp.sum(a=x, axis=axis, dtype=dtype, keepdims=keepdims)