Add jnp.get_printoptions and jnp.printoptions.

`jnp.set_printoptions` already exists but other printing utilities do not.

`jnp.get_printoptions` forwards `numpy.get_printoptions`.
`jnp.printoptions` forwards `numpy.printoptions`.

docs/jax.numpy.rst

@@ -177,6 +177,7 @@ Not every function in NumPy is implemented; contributions are welcome!
     full
     full_like
     gcd
+    get_printoptions
     geomspace
     gradient
     greater
@@ -301,6 +302,7 @@ Not every function in NumPy is implemented; contributions are welcome!
     polyval
     positive
     power
+    printoptions
     prod
     product
     promote_types

jax/_src/numpy/lax_numpy.py

@@ -89,7 +89,10 @@ PZERO = np.PZERO
 NZERO = np.NZERO
 nan = np.nan
 
-# And some numpy utility functions
+# NumPy utility functions
+
+get_printoptions = np.get_printoptions
+printoptions = np.printoptions
 set_printoptions = np.set_printoptions
 
 # We want isinstance(x, np.ndarray) checks in user code to work with the our

jax/numpy/__init__.py

@@ -161,6 +161,7 @@ from jax._src.numpy.lax_numpy import (
     full_like as full_like,
     gcd as gcd,
     geomspace as geomspace,
+    get_printoptions as get_printoptions,
     gradient as gradient,
     greater as greater,
     greater_equal as greater_equal,
@@ -289,6 +290,7 @@ from jax._src.numpy.lax_numpy import (
     polyval as polyval,
     positive as positive,
     power as power,
+    printoptions as printoptions,
     prod as prod,
     product as product,
     promote_types as promote_types,