jax.numpy: improved errors for invalid inputs to unary ops
jax.numpy: improved errors for invalid inputs to unary ops

google/jax,google/jax,tensorflow/probability,tensorflow/probability,google/jax,google/jax