Various type inconsistencies
Please consider adding `jax.numpy.generic` as an alias for `numpy.generic` (it seems to be missing).  Also,
```
In [5]: jnp.issubdtype(jnp.bfloat16, jnp.inexact)
Out[5]: True

In [6]: issubclass(jnp.bfloat16, jnp.inexact)
Out[6]: False

In [7]: issubclass(np.float32, np.inexact)
Out[7]: True
```
Line 6 should probably return true, which means somehow registering the types like numpy does.
diff --git a/jax/_src/numpy/lax_numpy.py b/jax/_src/numpy/lax_numpy.py
index 6e4e44b80..979478513 100644
--- a/jax/_src/numpy/lax_numpy.py
+++ b/jax/_src/numpy/lax_numpy.py
@@ -424,6 +424,7 @@ uint = uint32 if dtypes.uint == np.uint32 else uint64
 float_ = float32 if dtypes.float_ == np.float32 else float64
 complex_ = complex64 if dtypes.complex_ == np.complex64 else complex128
 
+generic = np.generic
 number = np.number
 inexact = np.inexact
 complexfloating = np.complexfloating
diff --git a/jax/numpy/__init__.py b/jax/numpy/__init__.py
index f28ec37c5..d51dc6a19 100644
--- a/jax/numpy/__init__.py
+++ b/jax/numpy/__init__.py
@@ -160,6 +160,7 @@ from jax._src.numpy.lax_numpy import (
     full as full,
     full_like as full_like,
     gcd as gcd,
+    generic as generic,
     geomspace as geomspace,
     get_printoptions as get_printoptions,
     gradient as gradient,
['jax/_src/numpy/lax_numpy.py', 'jax/numpy/__init__.py']