Contents:
Returns if x is a scalar
x (jax.numpy.ndarray) – A JAX array.
True if x is a scalar quantity (i.e. ndim==0).