Fix lax_reference implementation of round() to match lax. (#2894)
Fix lax_reference implementation of round() to match lax. (#2894)

lax.round() is documented to round half away from zero, but np.round() rounds to nearest even.
google/jax,google/jax,google/jax,google/jax,tensorflow/probability,tensorflow/probability