From d546503c740ef731781b347b822897ef1d1a312b Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Tue, 31 May 2022 20:56:46 -0400 Subject: [PATCH] Remove experimental warning from Mac ARM wheels. Enough JAX developers now have Mac ARM machines that we can be reasonably confident of noticing problems, even if we don't yet have CI coverage. --- jax/_src/lib/__init__.py | 8 -------- 1 file changed, 8 deletions(-) diff --git a/jax/_src/lib/__init__.py b/jax/_src/lib/__init__.py index f45bf729551d..03802ef01f09 100644 --- a/jax/_src/lib/__init__.py +++ b/jax/_src/lib/__init__.py @@ -28,14 +28,6 @@ 'tpu_driver_client', 'version', 'xla_client', 'xla_extension', ] -# Before attempting to import jaxlib, warn about experimental -# machine configurations. -if platform.system() == "Darwin" and platform.machine() == "arm64": - warnings.warn("JAX on Mac ARM machines is experimental and minimally tested. " - "Please see https://github.com/google/jax/issues/5501 in the " - "event of problems.") - - # This apparently-unused import is to work around # https://github.com/google/jax/issues/9218#issuecomment-1016949739 # If the user is using Conda, we want to ensure that Conda's libstdc++ is chosen