---------------------------------------------------------------------------
ImportError                               Traceback (most recent call last)
Cell In[1], line 3
      1 get_ipython().run_line_magic("matplotlib", "qt5")  # noqa: F821
----> 3 import jax

File C:\...\Lib\site-packages\jax\__init__.py:25
     22 from jax.version import __version_info__ as __version_info__
     24 # Set Cloud TPU env vars if necessary before transitively loading C++ backend
---> 25 from jax._src.cloud_tpu_init import cloud_tpu_init as _cloud_tpu_init
     26 try:
     27   _cloud_tpu_init()

File C:\...\Lib\site-packages\jax\_src\cloud_tpu_init.py:20
     17 import re
     18 import warnings
---> 20 from jax._src import config
     21 from jax._src import hardware_utils
     23 running_in_cloud_tpu_vm: bool = False

File C:\...\Lib\site-packages\jax\_src\config.py:29
     26 import warnings
     28 from jax._src import deprecations
---> 29 from jax._src import logging_config
     30 from jax._src.lib import _jax
     31 from jax._src.lib import guard_lib

File C:\...\Lib\site-packages\jax\_src\logging_config.py:17
     15 import logging
     16 import sys
---> 17 from jax._src.lib import utils
     19 # Example log message:
     20 # DEBUG:2023-06-07 00:14:40,280:jax._src.xla_bridge:590: Initializing backend 'cpu'
     21 logging_formatter = logging.Formatter(
     22     "{levelname}:{asctime}:{name}:{lineno}: {message}", style='{')

File C:\...\Lib\site-packages\jax\_src\lib\__init__.py:89
     86 import jaxlib.cpu_feature_guard as cpu_feature_guard
     87 cpu_feature_guard.check_cpu_features()
---> 89 import jaxlib.xla_client as xla_client  # noqa: F401
     91 # Jaxlib code is split between the Jax and the XLA repositories.
     92 # Only for the internal usage of the JAX developers, we expose a version
     93 # number that can be used to perform changes without breaking the main
     94 # branch on the Jax github.
     95 jaxlib_extension_version: int = getattr(xla_client, '_version', 0)

File C:\...\Lib\site-packages\jaxlib\xla_client.py:28
     25 import threading
     26 from typing import Any, Protocol, Union
---> 28 from jaxlib import _jax as _xla
     30 # Note this module does *not* depend on any Python protocol buffers. The XLA
     31 # Python bindings are currently packaged both as part of jaxlib and as part
     32 # of TensorFlow. If we use protocol buffers here, then importing both jaxlib
   (...)     39 # Pylint has false positives for type annotations.
     40 # pylint: disable=invalid-sequence-index
     42 ifrt_programs = _xla.ifrt_programs

ImportError: DLL load failed while importing _jax: A dynamic link library (DLL) initialization routine failed.