fix jax errors

Author: JuliaPoo

array_api_compat/jax/__init__.py

@@ -26,6 +26,7 @@
     # functions
     zeros,
     all,
+    any,
     isnan,
     isfinite,
     reshape
@@ -81,4 +82,4 @@ def top_k(
            'int64', 'uint8', 'uint16', 'uint32', 'uint64',
            'complex64', 'complex128', 'iinfo', 'finfo',
            'can_cast', 'result_type', 'zeros', 'all', 'isnan',
-           'isfinite', 'reshape']
+           'isfinite', 'reshape', 'any']