Bump minimum jaxlib/jax version

CHANGES.rst

@@ -20,7 +20,7 @@ Version 0.0.6   (unreleased)
 • Rename ``scico.flax.save_weights`` and ``scico.flax.load_weights`` to
   ``scico.flax.save_variables`` and ``scico.flax.load_variables``
   respectively.
-• Support ``jaxlib`` and ``jax`` versions 0.4.3 to 0.4.35.
+• Support ``jaxlib`` and ``jax`` versions 0.4.13 to 0.4.35.
 • Support ``flax`` versions 0.8.0 to 0.9.0.
 
 

requirements.txt

@@ -4,8 +4,8 @@ scipy>=1.6.0
 imageio>=2.17
 tifffile
 matplotlib
-jaxlib>=0.4.3,<=0.4.35
-jax>=0.4.3,<=0.4.35
+jaxlib>=0.4.13,<=0.4.35
+jax>=0.4.13,<=0.4.35
 orbax-checkpoint>=0.5.0
 flax>=0.8.0,<=0.9.0
 pyabel>=0.9.0

scico/__init__.py

@@ -1,4 +1,4 @@
-# Copyright (C) 2021-2023 by SCICO Developers
+# Copyright (C) 2021-2024 by SCICO Developers
 # All rights reserved. BSD 3-clause License.
 # This file is part of the SCICO package. Details of the copyright and
 # user license can be found in the 'LICENSE' file distributed with the
@@ -16,14 +16,7 @@
 # isort: off
 
 # Suppress jax device warning. See https://github.com/google/jax/issues/6805
-# This only works for jax>0.3.23; for earlier versions, the getLogger
-# argument should be "absl". Two filters are included here due to a change
-# in jax between versions 0.4.2 and 0.4.8, both of which are supported by
-# scico.
-logging.getLogger("jax._src.lib.xla_bridge").addFilter(  # jax 0.4.2
-    logging.Filter("No GPU/TPU found, falling back to CPU.")
-)
-logging.getLogger("jax._src.xla_bridge").addFilter(  # jax 0.4.8
+logging.getLogger("jax._src.xla_bridge").addFilter(  # jax 0.4.8 and later
     logging.Filter("No GPU/TPU found, falling back to CPU.")
 )