nit

Author: crusaderky

src/array_api_extra/_lib/_compat.pyi

@@ -20,8 +20,8 @@ def array_namespace(
 def device(x: Array, /) -> Device: ...
 def is_cupy_namespace(x: object, /) -> bool: ...
 def is_dask_namespace(xp: ModuleType, /) -> bool: ...
-def is_jax_namespace(xp: ModuleType, /) -> bool: ...
 def is_jax_array(x: object, /) -> bool: ...
+def is_jax_namespace(xp: ModuleType, /) -> bool: ...
 def is_pydata_sparse_namespace(x: object, /) -> bool: ...
 def is_torch_namespace(x: object, /) -> bool: ...
 def is_writeable_array(x: object, /) -> bool: ...