jax

063581a3 - Small optimization when forming IR constant.

Small optimization when forming IR constant. It's faster and simpler to write `0 in x.strides` than `np.any(np.equal(0, x.strides))`. No behavior changes intended. PiperOrigin-RevId: 648539396