Fix version check

Author: asmeurer

tests/_helpers.py

@@ -8,7 +8,7 @@
 def import_(library, wrapper=False):
     if library == 'cupy':
         return pytest.importorskip(library)
-    if 'jax' in library and sys.version_info <= (3, 8):
+    if 'jax' in library and sys.version_info < (3, 9):
         pytest.skip('JAX array API support does not support Python 3.8')
 
     if wrapper: