<p dir="auto">Hello,</p>
<p dir="auto">I just installed jax and jaxlib under macOS 11.1 (Big Sur) and Python 3.8.5, and issued a <code class="notranslate">pip install --upgrade pip tensorflow jax jaxlib</code> to start playing around with jax. At first attempt, <code class="notranslate">import jax</code> fails:</p>
<div class="highlight highlight-source-python notranslate position-relative overflow-auto" dir="auto" data-snippet-clipboard-copy-content="ImportError: dlopen(/Users/lucchem/opt/miniconda3/lib/python3.8/site-packages/jaxlib/xla_extension.so, 2): Symbol not found: __ZN10tensorflow8profiler11MonitorGrpcERKNSt3__112basic_stringIcNS1_11char_traitsIcEENS1_9allocatorIcEEEERKNS_14MonitorRequestEPNS_15MonitorResponseE
  Referenced from: /Users/lucchem/opt/miniconda3/lib/python3.8/site-packages/jaxlib/xla_extension.so
  Expected in: flat namespace
 in /Users/lucchem/opt/miniconda3/lib/python3.8/site-packages/jaxlib/xla_extension.so"><pre class="notranslate"><span class="pl-v">ImportError</span>: <span class="pl-s1">dlopen</span>(<span class="pl-c1">/</span><span class="pl-v">Users</span><span class="pl-c1">/</span><span class="pl-s1">lucchem</span><span class="pl-c1">/</span><span class="pl-s1">opt</span><span class="pl-c1">/</span><span class="pl-s1">miniconda3</span><span class="pl-c1">/</span><span class="pl-s1">lib</span><span class="pl-c1">/</span><span class="pl-s1">python3</span>.<span class="pl-c1">8</span><span class="pl-c1">/</span><span class="pl-s1">site</span><span class="pl-c1">-</span><span class="pl-s1">packages</span><span class="pl-c1">/</span><span class="pl-s1">jaxlib</span><span class="pl-c1">/</span><span class="pl-s1">xla_extension</span>.<span class="pl-s1">so</span>, <span class="pl-c1">2</span>): <span class="pl-v">Symbol</span> <span class="pl-c1">not</span> <span class="pl-s1">found</span>: <span class="pl-s1">__ZN10tensorflow8profiler11MonitorGrpcERKNSt3__112basic_stringIcNS1_11char_traitsIcEENS1_9allocatorIcEEEERKNS_14MonitorRequestEPNS_15MonitorResponseE</span>
  <span class="pl-v">Referenced</span> <span class="pl-s1">from</span>: <span class="pl-c1">/</span><span class="pl-v">Users</span><span class="pl-c1">/</span><span class="pl-s1">lucchem</span><span class="pl-c1">/</span><span class="pl-s1">opt</span><span class="pl-c1">/</span><span class="pl-s1">miniconda3</span><span class="pl-c1">/</span><span class="pl-s1">lib</span><span class="pl-c1">/</span><span class="pl-s1">python3</span>.<span class="pl-c1">8</span><span class="pl-c1">/</span><span class="pl-s1">site</span><span class="pl-c1">-</span><span class="pl-s1">packages</span><span class="pl-c1">/</span><span class="pl-s1">jaxlib</span><span class="pl-c1">/</span><span class="pl-s1">xla_extension</span>.<span class="pl-s1">so</span>
  <span class="pl-v">Expected</span> <span class="pl-c1">in</span>: <span class="pl-s1">flat</span> <span class="pl-s1">namespace</span>
 <span class="pl-s1">in</span> <span class="pl-c1">/</span><span class="pl-v">Users</span><span class="pl-c1">/</span><span class="pl-s1">lucchem</span><span class="pl-c1">/</span><span class="pl-s1">opt</span><span class="pl-c1">/</span><span class="pl-s1">miniconda3</span><span class="pl-c1">/</span><span class="pl-s1">lib</span><span class="pl-c1">/</span><span class="pl-s1">python3</span>.<span class="pl-c1">8</span><span class="pl-c1">/</span><span class="pl-s1">site</span><span class="pl-c1">-</span><span class="pl-s1">packages</span><span class="pl-c1">/</span><span class="pl-s1">jaxlib</span><span class="pl-c1">/</span><span class="pl-s1">xla_extension</span>.<span class="pl-s1">so</span></pre></div>
<p dir="auto"><code class="notranslate">pip show pip jax jaxlib tensorflow</code> outputs:</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="Name: pip
Version: 21.0.1
...
---
Name: jax
Version: 0.2.10
...
---
Name: jaxlib
Version: 0.1.63
...
---
Name: tensorflow
Version: 2.4.1
..."><pre class="notranslate"><code class="notranslate">Name: pip
Version: 21.0.1
...
---
Name: jax
Version: 0.2.10
...
---
Name: jaxlib
Version: 0.1.63
...
---
Name: tensorflow
Version: 2.4.1
...
</code></pre></div>
<p dir="auto">is there any fix? Thank you in advance.</p>
<p dir="auto">Upgrading from jaxlib 0.1.62 to 0.1.63 causes import of jax to fail with mysterious error messages. Downgrading back to 0.1.62 and everything is happy again. (This in on Mac OS)</p>
<div class="snippet-clipboard-content notranslate position-relative overflow-auto" data-snippet-clipboard-copy-content="Requirement already satisfied: jaxlib in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (0.1.62)
Collecting jaxlib
  Using cached jaxlib-0.1.63-cp38-none-macosx_10_9_x86_64.whl (42.7 MB)
Requirement already satisfied: absl-py in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (0.12.0)
Requirement already satisfied: numpy&gt;=1.16 in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (1.20.1)
Requirement already satisfied: flatbuffers in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (20210203211748)
Requirement already satisfied: scipy in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (1.6.1)
Requirement already satisfied: six in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from absl-py-&gt;jaxlib) (1.15.0)
Installing collected packages: jaxlib
  Attempting uninstall: jaxlib
    Found existing installation: jaxlib 0.1.62
    Uninstalling jaxlib-0.1.62:
      Successfully uninstalled jaxlib-0.1.62
Successfully installed jaxlib-0.1.63

$ python -c &quot;import jax&quot;
Traceback (most recent call last):
  File &quot;&lt;string&gt;&quot;, line 1, in &lt;module&gt;
  File &quot;/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jax/__init__.py&quot;, line 21, in &lt;module&gt;
    from .config import config
  File &quot;/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jax/config.py&quot;, line 19, in &lt;module&gt;
    from jax import lib
  File &quot;/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jax/lib/__init__.py&quot;, line 53, in &lt;module&gt;
    from jaxlib import xla_client
  File &quot;/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_client.py&quot;, line 31, in &lt;module&gt;
    from . import xla_extension as _xla
ImportError: dlopen(/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_extension.so, 2): Symbol not found: __ZN10tensorflow8profiler11MonitorGrpcERKNSt3__112basic_stringIcNS1_11char_traitsIcEENS1_9allocatorIcEEEERKNS_14MonitorRequestEPNS_15MonitorResponseE
  Referenced from: /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_extension.so
  Expected in: flat namespace
 in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_extension.so"><pre lang="$" class="notranslate"><code class="notranslate">Requirement already satisfied: jaxlib in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (0.1.62)
Collecting jaxlib
  Using cached jaxlib-0.1.63-cp38-none-macosx_10_9_x86_64.whl (42.7 MB)
Requirement already satisfied: absl-py in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (0.12.0)
Requirement already satisfied: numpy&gt;=1.16 in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (1.20.1)
Requirement already satisfied: flatbuffers in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (20210203211748)
Requirement already satisfied: scipy in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from jaxlib) (1.6.1)
Requirement already satisfied: six in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages (from absl-py-&gt;jaxlib) (1.15.0)
Installing collected packages: jaxlib
  Attempting uninstall: jaxlib
    Found existing installation: jaxlib 0.1.62
    Uninstalling jaxlib-0.1.62:
      Successfully uninstalled jaxlib-0.1.62
Successfully installed jaxlib-0.1.63

$ python -c "import jax"
Traceback (most recent call last):
  File "&lt;string&gt;", line 1, in &lt;module&gt;
  File "/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jax/__init__.py", line 21, in &lt;module&gt;
    from .config import config
  File "/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jax/config.py", line 19, in &lt;module&gt;
    from jax import lib
  File "/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jax/lib/__init__.py", line 53, in &lt;module&gt;
    from jaxlib import xla_client
  File "/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_client.py", line 31, in &lt;module&gt;
    from . import xla_extension as _xla
ImportError: dlopen(/Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_extension.so, 2): Symbol not found: __ZN10tensorflow8profiler11MonitorGrpcERKNSt3__112basic_stringIcNS1_11char_traitsIcEENS1_9allocatorIcEEEERKNS_14MonitorRequestEPNS_15MonitorResponseE
  Referenced from: /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_extension.so
  Expected in: flat namespace
 in /Users/work/miniconda3/envs/QPACK/lib/python3.8/site-packages/jaxlib/xla_extension.so
</code></pre></div>