<p dir="auto">Please:</p>
<ul class="contains-task-list">
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox"> Check for duplicate requests.</li>
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox"> Describe your goal, and if possible provide a code snippet with a motivating example.<br>
Added support for python latest version 3.10</li>
</ul>
<ul class="contains-task-list">
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> Check for duplicate issues.</li>
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> Provide a complete example of how to reproduce the bug, wrapped in triple backticks like this:</li>
</ul>
<p dir="auto">Here's how to reproduce (Ubuntu 20.04, Intel Xeon 1650 CPU):</p>
<div class="highlight highlight-source-shell notranslate position-relative overflow-auto" dir="auto" data-snippet-clipboard-copy-content="# Create a fresh virtual env
conda create -n venv python=3.9
conda activate venv

# Install `jax` from conda
conda install jax

# Since I wanted GPU support, next line only removes `jaxlib`, not other deps
conda remove jaxlib --force

# Now I install GPU-compatible `jaxlib`
pip install --upgrade &quot;jax[cuda]&quot; -f https://storage.googleapis.com/jax-releases/jax_releases.html"><pre class="notranslate"><span class="pl-c"><span class="pl-c">#</span> Create a fresh virtual env</span>
conda create -n venv python=3.9
conda activate venv

<span class="pl-c"><span class="pl-c">#</span> Install `jax` from conda</span>
conda install jax

<span class="pl-c"><span class="pl-c">#</span> Since I wanted GPU support, next line only removes `jaxlib`, not other deps</span>
conda remove jaxlib --force

<span class="pl-c"><span class="pl-c">#</span> Now I install GPU-compatible `jaxlib`</span>
pip install --upgrade <span class="pl-s"><span class="pl-pds">"</span>jax[cuda]<span class="pl-pds">"</span></span> -f https://storage.googleapis.com/jax-releases/jax_releases.html</pre></div>
<p dir="auto">Now, just enter Python REPL, and import <code class="notranslate">jax</code>:</p>
<div class="highlight highlight-source-python notranslate position-relative overflow-auto" dir="auto" data-snippet-clipboard-copy-content="&gt;&gt;&gt; import jax"><pre class="notranslate"><span class="pl-c1">&gt;&gt;</span><span class="pl-c1">&gt;</span> <span class="pl-k">import</span> <span class="pl-s1">jax</span></pre></div>
<ul class="contains-task-list">
<li class="task-list-item"><input type="checkbox" id="" disabled="" class="task-list-item-checkbox" checked=""> If applicable, include full error messages/tracebacks.</li>
</ul>
<div class="highlight highlight-source-python notranslate position-relative overflow-auto" dir="auto" data-snippet-clipboard-copy-content="Traceback (most recent call last):
  File &quot;/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/scipy/linalg/__init__.py&quot;, line 210, in &lt;module&gt;
    from ._matfuncs import *
  File &quot;/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/scipy/linalg/_matfuncs.py&quot;, line 21, in &lt;module&gt;
    from ._matfuncs_sqrtm import sqrtm
  File &quot;/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/scipy/linalg/_matfuncs_sqrtm.py&quot;, line 24, in &lt;module&gt;
    from ._matfuncs_sqrtm_triu import within_block_loop
ImportError: /usr/lib/x86_64-linux-gnu/libstdc++.so.6: version `GLIBCXX_3.4.26' not found (required by /home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/scipy/linalg/_matfuncs_sqrtm_triu.cpython-39-x86_64-linux-gnu.so)

The above exception was the direct cause of the following exception:

Traceback (most recent call last):
  File &quot;&lt;stdin&gt;&quot;, line 1, in &lt;module&gt;
  File &quot;/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/jax/__init__.py&quot;, line 37, in &lt;module&gt;
    from jax import config as _config_module
  File &quot;/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/jax/config.py&quot;, line 18, in &lt;module&gt;
    from jax._src.config import config
  File &quot;/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/jax/_src/config.py&quot;, line 27, in &lt;module&gt;
    from jax._src import lib
  File &quot;/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/jax/_src/lib/__init__.py&quot;, line 101, in &lt;module&gt;
    import jaxlib.lapack as lapack
  File &quot;/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/jaxlib/lapack.py&quot;, line 21, in &lt;module&gt;
    from . import _lapack
ImportError: initialization failed"><pre class="notranslate"><span class="pl-v">Traceback</span> (<span class="pl-s1">most</span> <span class="pl-s1">recent</span> <span class="pl-s1">call</span> <span class="pl-s1">last</span>):
  <span class="pl-v">File</span> <span class="pl-s">"/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/scipy/linalg/__init__.py"</span>, <span class="pl-s1">line</span> <span class="pl-c1">210</span>, <span class="pl-s1">in</span> <span class="pl-c1">&lt;</span><span class="pl-s1">module</span><span class="pl-c1">&gt;</span>
    <span class="pl-s1">from</span> .<span class="pl-s1">_matfuncs</span> <span class="pl-s1">import</span> <span class="pl-c1">*</span>
  <span class="pl-v">File</span> <span class="pl-s">"/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/scipy/linalg/_matfuncs.py"</span>, <span class="pl-s1">line</span> <span class="pl-c1">21</span>, <span class="pl-s1">in</span> <span class="pl-c1">&lt;</span><span class="pl-s1">module</span><span class="pl-c1">&gt;</span>
    <span class="pl-s1">from</span> .<span class="pl-s1">_matfuncs_sqrtm</span> <span class="pl-k">import</span> <span class="pl-s1">sqrtm</span>
  <span class="pl-v">File</span> <span class="pl-s">"/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/scipy/linalg/_matfuncs_sqrtm.py"</span>, <span class="pl-s1">line</span> <span class="pl-c1">24</span>, <span class="pl-s1">in</span> <span class="pl-c1">&lt;</span><span class="pl-s1">module</span><span class="pl-c1">&gt;</span>
    <span class="pl-s1">from</span> .<span class="pl-s1">_matfuncs_sqrtm_triu</span> <span class="pl-k">import</span> <span class="pl-s1">within_block_loop</span>
<span class="pl-v">ImportError</span>: <span class="pl-c1">/</span><span class="pl-s1">usr</span><span class="pl-c1">/</span><span class="pl-s1">lib</span><span class="pl-c1">/</span><span class="pl-s1">x86_64</span><span class="pl-c1">-</span><span class="pl-s1">linux</span><span class="pl-c1">-</span><span class="pl-s1">gnu</span><span class="pl-c1">/</span><span class="pl-s1">libstdc</span><span class="pl-c1">+</span><span class="pl-c1">+</span>.<span class="pl-s1">so</span>.<span class="pl-c1">6</span>: <span class="pl-s1">version</span> `<span class="pl-v">GLIBCXX_3</span><span class="pl-c1">.4</span>.<span class="pl-c1">26</span>' <span class="pl-s1">not</span> <span class="pl-en">found</span> (<span class="pl-s1">required</span> <span class="pl-s1">by</span> <span class="pl-c1">/</span><span class="pl-s1">home</span><span class="pl-c1">/</span><span class="pl-s1">anaconda</span><span class="pl-c1">/</span><span class="pl-s1">mambaforge</span><span class="pl-c1">/</span><span class="pl-s1">envs</span><span class="pl-c1">/</span><span class="pl-s1">gpu</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">9</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">scipy</span><span class="pl-c1">/</span><span class="pl-s1">linalg</span><span class="pl-c1">/</span><span class="pl-s1">_matfuncs_sqrtm_triu</span>.<span class="pl-s1">cpython</span><span class="pl-c1">-</span><span class="pl-c1">39</span><span class="pl-c1">-</span><span class="pl-s1">x86_64</span><span class="pl-c1">-</span><span class="pl-s1">linux</span><span class="pl-c1">-</span><span class="pl-s1">gnu</span>.<span class="pl-s1">so</span>)

<span class="pl-v">The</span> <span class="pl-s1">above</span> <span class="pl-s1">exception</span> <span class="pl-s1">was</span> <span class="pl-s1">the</span> <span class="pl-s1">direct</span> <span class="pl-s1">cause</span> <span class="pl-s1">of</span> <span class="pl-s1">the</span> <span class="pl-s1">following</span> <span class="pl-s1">exception</span>:

<span class="pl-v">Traceback</span> (<span class="pl-s1">most</span> <span class="pl-s1">recent</span> <span class="pl-s1">call</span> <span class="pl-s1">last</span>):
  <span class="pl-v">File</span> <span class="pl-s">"&lt;stdin&gt;"</span>, <span class="pl-s1">line</span> <span class="pl-c1">1</span>, <span class="pl-c1">in</span> <span class="pl-c1">&lt;</span><span class="pl-s1">module</span><span class="pl-c1">&gt;</span>
  <span class="pl-v">File</span> <span class="pl-s">"/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/jax/__init__.py"</span>, <span class="pl-s1">line</span> <span class="pl-c1">37</span>, <span class="pl-s1">in</span> <span class="pl-c1">&lt;</span><span class="pl-s1">module</span><span class="pl-c1">&gt;</span>
    <span class="pl-s1">from</span> <span class="pl-s1">jax</span> <span class="pl-k">import</span> <span class="pl-s1">config</span> <span class="pl-k">as</span> <span class="pl-s1">_config_module</span>
  <span class="pl-v">File</span> <span class="pl-s">"/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/jax/config.py"</span>, <span class="pl-s1">line</span> <span class="pl-c1">18</span>, <span class="pl-s1">in</span> <span class="pl-c1">&lt;</span><span class="pl-s1">module</span><span class="pl-c1">&gt;</span>
    <span class="pl-s1">from</span> <span class="pl-s1">jax</span>.<span class="pl-s1">_src</span>.<span class="pl-s1">config</span> <span class="pl-k">import</span> <span class="pl-s1">config</span>
  <span class="pl-v">File</span> <span class="pl-s">"/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/jax/_src/config.py"</span>, <span class="pl-s1">line</span> <span class="pl-c1">27</span>, <span class="pl-s1">in</span> <span class="pl-c1">&lt;</span><span class="pl-s1">module</span><span class="pl-c1">&gt;</span>
    <span class="pl-s1">from</span> <span class="pl-s1">jax</span>.<span class="pl-s1">_src</span> <span class="pl-k">import</span> <span class="pl-s1">lib</span>
  <span class="pl-v">File</span> <span class="pl-s">"/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/jax/_src/lib/__init__.py"</span>, <span class="pl-s1">line</span> <span class="pl-c1">101</span>, <span class="pl-s1">in</span> <span class="pl-c1">&lt;</span><span class="pl-s1">module</span><span class="pl-c1">&gt;</span>
    <span class="pl-s1">import</span> <span class="pl-s1">jaxlib</span>.<span class="pl-s1">lapack</span> <span class="pl-k">as</span> <span class="pl-s1">lapack</span>
  <span class="pl-v">File</span> <span class="pl-s">"/home/anaconda/mambaforge/envs/gpu/lib/python3.9/site-packages/jaxlib/lapack.py"</span>, <span class="pl-s1">line</span> <span class="pl-c1">21</span>, <span class="pl-s1">in</span> <span class="pl-c1">&lt;</span><span class="pl-s1">module</span><span class="pl-c1">&gt;</span>
    <span class="pl-s1">from</span> . <span class="pl-s1">import</span> <span class="pl-s1">_lapack</span>
<span class="pl-v">ImportError</span>: <span class="pl-s1">initialization</span> <span class="pl-s1">failed</span></pre></div>
<p dir="auto">The reason that in the script above I initially install <code class="notranslate">jax</code> from <code class="notranslate">conda</code> although it's not GPU-compatible is because I want the packages to mostly come from <code class="notranslate">conda-forge</code> (for compatibility reasons) and only if that's not possible, get them from <code class="notranslate">pypi</code>. In this case, I only want to install GPU-compatible <code class="notranslate">jaxlib</code> from PyPI and the rest from <code class="notranslate">conda-forge</code>.</p>
<p dir="auto">If it helps debugging the issue, I tried getting all dependencies from <code class="notranslate">pypi</code> (essentially installing <code class="notranslate">jax</code> from <code class="notranslate">pypi</code> right off the bat after creating a new virtual env), and it works fine. so there must be some package from <code class="notranslate">conda-forge</code> that causes the <code class="notranslate">ImportError</code>.</p>
<p dir="auto">To further illustrate the issue, here's the output of <code class="notranslate">conda list</code> when everything comes from <code class="notranslate">pypi</code>:</p>
<div class="highlight highlight-source-shell notranslate position-relative overflow-auto" dir="auto" data-snippet-clipboard-copy-content="# Name                    Version                   Build  Channel
_libgcc_mutex             0.1                 conda_forge    conda-forge
_openmp_mutex             4.5                       1_gnu    conda-forge
absl-py                   1.0.0                    pypi_0    pypi
bzip2                     1.0.8                h7f98852_4    conda-forge
ca-certificates           2021.10.8            ha878542_0    conda-forge
flatbuffers               2.0                      pypi_0    pypi
jax                       0.3.4                    pypi_0    pypi
jaxlib                    0.3.2+cuda11.cudnn82          pypi_0    pypi
ld_impl_linux-64          2.36.1               hea4e1c9_2    conda-forge
libffi                    3.4.2                h7f98852_5    conda-forge
libgcc-ng                 11.2.0              h1d223b6_14    conda-forge
libgomp                   11.2.0              h1d223b6_14    conda-forge
libnsl                    2.0.0                h7f98852_0    conda-forge
libuuid                   2.32.1            h7f98852_1000    conda-forge
libzlib                   1.2.11            h166bdaf_1014    conda-forge
ncurses                   6.3                  h9c3ff4c_0    conda-forge
numpy                     1.22.3                   pypi_0    pypi
openssl                   3.0.2                h166bdaf_1    conda-forge
opt-einsum                3.3.0                    pypi_0    pypi
pip                       22.0.4             pyhd8ed1ab_0    conda-forge
python                    3.9.12          h2660328_1_cpython    conda-forge
python_abi                3.9                      2_cp39    conda-forge
readline                  8.1                  h46c0cb4_0    conda-forge
scipy                     1.8.0                    pypi_0    pypi
setuptools                61.2.0           py39hf3d152e_1    conda-forge
six                       1.16.0                   pypi_0    pypi
sqlite                    3.37.1               h4ff8645_0    conda-forge
tk                        8.6.12               h27826a3_0    conda-forge
typing-extensions         4.1.1                    pypi_0    pypi
tzdata                    2022a                h191b570_0    conda-forge
wheel                     0.37.1             pyhd8ed1ab_0    conda-forge
xz                        5.2.5                h516909a_1    conda-forge
zlib                      1.2.11            h166bdaf_1014    conda-forge"><pre class="notranslate"><span class="pl-c"><span class="pl-c">#</span> Name                    Version                   Build  Channel</span>
_libgcc_mutex             0.1                 conda_forge    conda-forge
_openmp_mutex             4.5                       1_gnu    conda-forge
absl-py                   1.0.0                    pypi_0    pypi
bzip2                     1.0.8                h7f98852_4    conda-forge
ca-certificates           2021.10.8            ha878542_0    conda-forge
flatbuffers               2.0                      pypi_0    pypi
jax                       0.3.4                    pypi_0    pypi
jaxlib                    0.3.2+cuda11.cudnn82          pypi_0    pypi
ld_impl_linux-64          2.36.1               hea4e1c9_2    conda-forge
libffi                    3.4.2                h7f98852_5    conda-forge
libgcc-ng                 11.2.0              h1d223b6_14    conda-forge
libgomp                   11.2.0              h1d223b6_14    conda-forge
libnsl                    2.0.0                h7f98852_0    conda-forge
libuuid                   2.32.1            h7f98852_1000    conda-forge
libzlib                   1.2.11            h166bdaf_1014    conda-forge
ncurses                   6.3                  h9c3ff4c_0    conda-forge
numpy                     1.22.3                   pypi_0    pypi
openssl                   3.0.2                h166bdaf_1    conda-forge
opt-einsum                3.3.0                    pypi_0    pypi
pip                       22.0.4             pyhd8ed1ab_0    conda-forge
python                    3.9.12          h2660328_1_cpython    conda-forge
python_abi                3.9                      2_cp39    conda-forge
readline                  8.1                  h46c0cb4_0    conda-forge
scipy                     1.8.0                    pypi_0    pypi
setuptools                61.2.0           py39hf3d152e_1    conda-forge
six                       1.16.0                   pypi_0    pypi
sqlite                    3.37.1               h4ff8645_0    conda-forge
tk                        8.6.12               h27826a3_0    conda-forge
typing-extensions         4.1.1                    pypi_0    pypi
tzdata                    2022a                h191b570_0    conda-forge
wheel                     0.37.1             pyhd8ed1ab_0    conda-forge
xz                        5.2.5                h516909a_1    conda-forge
zlib                      1.2.11            h166bdaf_1014    conda-forge</pre></div>
<p dir="auto">and here's its output when only <code class="notranslate">jaxlib</code> is coming from <code class="notranslate">pypi</code>:</p>
<div class="highlight highlight-source-shell notranslate position-relative overflow-auto" dir="auto" data-snippet-clipboard-copy-content="# Name                    Version                   Build  Channel
_libgcc_mutex             0.1                 conda_forge    conda-forge
_openmp_mutex             4.5                       1_gnu    conda-forge
abseil-cpp                20211102.0           h27087fc_1    conda-forge
absl-py                   1.0.0              pyhd8ed1ab_0    conda-forge
bzip2                     1.0.8                h7f98852_4    conda-forge
c-ares                    1.18.1               h7f98852_0    conda-forge
ca-certificates           2021.10.8            ha878542_0    conda-forge
grpc-cpp                  1.44.0               h1021d7f_1    conda-forge
jax                       0.3.4              pyhd8ed1ab_0    conda-forge
jaxlib                    0.3.2+cuda11.cudnn82          pypi_0    pypi
ld_impl_linux-64          2.36.1               hea4e1c9_2    conda-forge
libblas                   3.9.0           13_linux64_openblas    conda-forge
libcblas                  3.9.0           13_linux64_openblas    conda-forge
libffi                    3.4.2                h7f98852_5    conda-forge
libgcc-ng                 11.2.0              h1d223b6_14    conda-forge
libgfortran-ng            11.2.0              h69a702a_14    conda-forge
libgfortran5              11.2.0              h5c6108e_14    conda-forge
libgomp                   11.2.0              h1d223b6_14    conda-forge
liblapack                 3.9.0           13_linux64_openblas    conda-forge
libnsl                    2.0.0                h7f98852_0    conda-forge
libopenblas               0.3.18          pthreads_h8fe5266_0    conda-forge
libprotobuf               3.19.4               h780b84a_0    conda-forge
libstdcxx-ng              11.2.0              he4da1e4_14    conda-forge
libuuid                   2.32.1            h7f98852_1000    conda-forge
libzlib                   1.2.11            h166bdaf_1014    conda-forge
ncurses                   6.3                  h9c3ff4c_0    conda-forge
numpy                     1.22.3           py39h18676bf_0    conda-forge
openssl                   3.0.2                h166bdaf_1    conda-forge
opt_einsum                3.3.0              pyhd8ed1ab_1    conda-forge
pip                       22.0.4             pyhd8ed1ab_0    conda-forge
python                    3.9.12          h2660328_1_cpython    conda-forge
python-flatbuffers        2.0                pyhd8ed1ab_0    conda-forge
python_abi                3.9                      2_cp39    conda-forge
re2                       2022.02.01           h9c3ff4c_0    conda-forge
readline                  8.1                  h46c0cb4_0    conda-forge
scipy                     1.8.0            py39hee8e79c_1    conda-forge
setuptools                61.2.0           py39hf3d152e_1    conda-forge
six                       1.16.0             pyh6c4a22f_0    conda-forge
sqlite                    3.37.1               h4ff8645_0    conda-forge
tk                        8.6.12               h27826a3_0    conda-forge
typing_extensions         4.1.1              pyha770c72_0    conda-forge
tzdata                    2022a                h191b570_0    conda-forge
wheel                     0.37.1             pyhd8ed1ab_0    conda-forge
xz                        5.2.5                h516909a_1    conda-forge
zlib                      1.2.11            h166bdaf_1014    conda-forge"><pre class="notranslate"><span class="pl-c"><span class="pl-c">#</span> Name                    Version                   Build  Channel</span>
_libgcc_mutex             0.1                 conda_forge    conda-forge
_openmp_mutex             4.5                       1_gnu    conda-forge
abseil-cpp                20211102.0           h27087fc_1    conda-forge
absl-py                   1.0.0              pyhd8ed1ab_0    conda-forge
bzip2                     1.0.8                h7f98852_4    conda-forge
c-ares                    1.18.1               h7f98852_0    conda-forge
ca-certificates           2021.10.8            ha878542_0    conda-forge
grpc-cpp                  1.44.0               h1021d7f_1    conda-forge
jax                       0.3.4              pyhd8ed1ab_0    conda-forge
jaxlib                    0.3.2+cuda11.cudnn82          pypi_0    pypi
ld_impl_linux-64          2.36.1               hea4e1c9_2    conda-forge
libblas                   3.9.0           13_linux64_openblas    conda-forge
libcblas                  3.9.0           13_linux64_openblas    conda-forge
libffi                    3.4.2                h7f98852_5    conda-forge
libgcc-ng                 11.2.0              h1d223b6_14    conda-forge
libgfortran-ng            11.2.0              h69a702a_14    conda-forge
libgfortran5              11.2.0              h5c6108e_14    conda-forge
libgomp                   11.2.0              h1d223b6_14    conda-forge
liblapack                 3.9.0           13_linux64_openblas    conda-forge
libnsl                    2.0.0                h7f98852_0    conda-forge
libopenblas               0.3.18          pthreads_h8fe5266_0    conda-forge
libprotobuf               3.19.4               h780b84a_0    conda-forge
libstdcxx-ng              11.2.0              he4da1e4_14    conda-forge
libuuid                   2.32.1            h7f98852_1000    conda-forge
libzlib                   1.2.11            h166bdaf_1014    conda-forge
ncurses                   6.3                  h9c3ff4c_0    conda-forge
numpy                     1.22.3           py39h18676bf_0    conda-forge
openssl                   3.0.2                h166bdaf_1    conda-forge
opt_einsum                3.3.0              pyhd8ed1ab_1    conda-forge
pip                       22.0.4             pyhd8ed1ab_0    conda-forge
python                    3.9.12          h2660328_1_cpython    conda-forge
python-flatbuffers        2.0                pyhd8ed1ab_0    conda-forge
python_abi                3.9                      2_cp39    conda-forge
re2                       2022.02.01           h9c3ff4c_0    conda-forge
readline                  8.1                  h46c0cb4_0    conda-forge
scipy                     1.8.0            py39hee8e79c_1    conda-forge
setuptools                61.2.0           py39hf3d152e_1    conda-forge
six                       1.16.0             pyh6c4a22f_0    conda-forge
sqlite                    3.37.1               h4ff8645_0    conda-forge
tk                        8.6.12               h27826a3_0    conda-forge
typing_extensions         4.1.1              pyha770c72_0    conda-forge
tzdata                    2022a                h191b570_0    conda-forge
wheel                     0.37.1             pyhd8ed1ab_0    conda-forge
xz                        5.2.5                h516909a_1    conda-forge
zlib                      1.2.11            h166bdaf_1014    conda-forge</pre></div>
<p dir="auto">and here's the diff between the two:</p>
<div class="highlight highlight-source-shell notranslate position-relative overflow-auto" dir="auto" data-snippet-clipboard-copy-content=" # Name                    Version                   Build  Channel
 _libgcc_mutex             0.1                 conda_forge    conda-forge
 _openmp_mutex             4.5                       1_gnu    conda-forge
-abseil-cpp                20211102.0           h27087fc_1    conda-forge
-absl-py                   1.0.0              pyhd8ed1ab_0    conda-forge
+absl-py                   1.0.0                    pypi_0    pypi
 bzip2                     1.0.8                h7f98852_4    conda-forge
-c-ares                    1.18.1               h7f98852_0    conda-forge
 ca-certificates           2021.10.8            ha878542_0    conda-forge
-grpc-cpp                  1.44.0               h1021d7f_1    conda-forge
-jax                       0.3.4              pyhd8ed1ab_0    conda-forge
+flatbuffers               2.0                      pypi_0    pypi
+jax                       0.3.4                    pypi_0    pypi
 jaxlib                    0.3.2+cuda11.cudnn82          pypi_0    pypi
 ld_impl_linux-64          2.36.1               hea4e1c9_2    conda-forge
-libblas                   3.9.0           13_linux64_openblas    conda-forge
-libcblas                  3.9.0           13_linux64_openblas    conda-forge
 libffi                    3.4.2                h7f98852_5    conda-forge
 libgcc-ng                 11.2.0              h1d223b6_14    conda-forge
-libgfortran-ng            11.2.0              h69a702a_14    conda-forge
-libgfortran5              11.2.0              h5c6108e_14    conda-forge
 libgomp                   11.2.0              h1d223b6_14    conda-forge
-liblapack                 3.9.0           13_linux64_openblas    conda-forge
 libnsl                    2.0.0                h7f98852_0    conda-forge
-libopenblas               0.3.18          pthreads_h8fe5266_0    conda-forge
-libprotobuf               3.19.4               h780b84a_0    conda-forge
-libstdcxx-ng              11.2.0              he4da1e4_14    conda-forge
 libuuid                   2.32.1            h7f98852_1000    conda-forge
 libzlib                   1.2.11            h166bdaf_1014    conda-forge
 ncurses                   6.3                  h9c3ff4c_0    conda-forge
-numpy                     1.22.3           py39h18676bf_0    conda-forge
+numpy                     1.22.3                   pypi_0    pypi
 openssl                   3.0.2                h166bdaf_1    conda-forge
-opt_einsum                3.3.0              pyhd8ed1ab_1    conda-forge
+opt-einsum                3.3.0                    pypi_0    pypi
 pip                       22.0.4             pyhd8ed1ab_0    conda-forge
 python                    3.9.12          h2660328_1_cpython    conda-forge
-python-flatbuffers        2.0                pyhd8ed1ab_0    conda-forge
 python_abi                3.9                      2_cp39    conda-forge
-re2                       2022.02.01           h9c3ff4c_0    conda-forge
 readline                  8.1                  h46c0cb4_0    conda-forge
-scipy                     1.8.0            py39hee8e79c_1    conda-forge
+scipy                     1.8.0                    pypi_0    pypi
 setuptools                61.2.0           py39hf3d152e_1    conda-forge
-six                       1.16.0             pyh6c4a22f_0    conda-forge
+six                       1.16.0                   pypi_0    pypi
 sqlite                    3.37.1               h4ff8645_0    conda-forge
 tk                        8.6.12               h27826a3_0    conda-forge
-typing_extensions         4.1.1              pyha770c72_0    conda-forge
+typing-extensions         4.1.1                    pypi_0    pypi
 tzdata                    2022a                h191b570_0    conda-forge
 wheel                     0.37.1             pyhd8ed1ab_0    conda-forge
 xz                        5.2.5                h516909a_1    conda-forge"><pre class="notranslate"> <span class="pl-c"><span class="pl-c">#</span> Name                    Version                   Build  Channel</span>
 _libgcc_mutex             0.1                 conda_forge    conda-forge
 _openmp_mutex             4.5                       1_gnu    conda-forge
-abseil-cpp                20211102.0           h27087fc_1    conda-forge
-absl-py                   1.0.0              pyhd8ed1ab_0    conda-forge
+absl-py                   1.0.0                    pypi_0    pypi
 bzip2                     1.0.8                h7f98852_4    conda-forge
-c-ares                    1.18.1               h7f98852_0    conda-forge
 ca-certificates           2021.10.8            ha878542_0    conda-forge
-grpc-cpp                  1.44.0               h1021d7f_1    conda-forge
-jax                       0.3.4              pyhd8ed1ab_0    conda-forge
+flatbuffers               2.0                      pypi_0    pypi
+jax                       0.3.4                    pypi_0    pypi
 jaxlib                    0.3.2+cuda11.cudnn82          pypi_0    pypi
 ld_impl_linux-64          2.36.1               hea4e1c9_2    conda-forge
-libblas                   3.9.0           13_linux64_openblas    conda-forge
-libcblas                  3.9.0           13_linux64_openblas    conda-forge
 libffi                    3.4.2                h7f98852_5    conda-forge
 libgcc-ng                 11.2.0              h1d223b6_14    conda-forge
-libgfortran-ng            11.2.0              h69a702a_14    conda-forge
-libgfortran5              11.2.0              h5c6108e_14    conda-forge
 libgomp                   11.2.0              h1d223b6_14    conda-forge
-liblapack                 3.9.0           13_linux64_openblas    conda-forge
 libnsl                    2.0.0                h7f98852_0    conda-forge
-libopenblas               0.3.18          pthreads_h8fe5266_0    conda-forge
-libprotobuf               3.19.4               h780b84a_0    conda-forge
-libstdcxx-ng              11.2.0              he4da1e4_14    conda-forge
 libuuid                   2.32.1            h7f98852_1000    conda-forge
 libzlib                   1.2.11            h166bdaf_1014    conda-forge
 ncurses                   6.3                  h9c3ff4c_0    conda-forge
-numpy                     1.22.3           py39h18676bf_0    conda-forge
+numpy                     1.22.3                   pypi_0    pypi
 openssl                   3.0.2                h166bdaf_1    conda-forge
-opt_einsum                3.3.0              pyhd8ed1ab_1    conda-forge
+opt-einsum                3.3.0                    pypi_0    pypi
 pip                       22.0.4             pyhd8ed1ab_0    conda-forge
 python                    3.9.12          h2660328_1_cpython    conda-forge
-python-flatbuffers        2.0                pyhd8ed1ab_0    conda-forge
 python_abi                3.9                      2_cp39    conda-forge
-re2                       2022.02.01           h9c3ff4c_0    conda-forge
 readline                  8.1                  h46c0cb4_0    conda-forge
-scipy                     1.8.0            py39hee8e79c_1    conda-forge
+scipy                     1.8.0                    pypi_0    pypi
 setuptools                61.2.0           py39hf3d152e_1    conda-forge
-six                       1.16.0             pyh6c4a22f_0    conda-forge
+six                       1.16.0                   pypi_0    pypi
 sqlite                    3.37.1               h4ff8645_0    conda-forge
 tk                        8.6.12               h27826a3_0    conda-forge
-typing_extensions         4.1.1              pyha770c72_0    conda-forge
+typing-extensions         4.1.1                    pypi_0    pypi
 tzdata                    2022a                h191b570_0    conda-forge
 wheel                     0.37.1             pyhd8ed1ab_0    conda-forge
 xz                        5.2.5                h516909a_1    conda-forge</pre></div>