jax

9b677fc1 - `is_fully_replicated` should return False if the underlying array is unreduced on any mesh axes.