Error "UnboundLocalError: local variable 'name' referenced before assignment" import jax.numpy as np · Issue #253 · jax-ml/jax (original) (raw)

I have installed JAX using pip: pip install --upgrade jax jaxlib # CPU-only version
(MACOS 10.14.2 and python 3.6.5)
Got error "UnboundLocalError: local variable 'name' referenced before assignment" when import jax.numpy as np

/anaconda3/lib/python3.6/importlib/_bootstrap.py:219: RuntimeWarning: compiletime version 3.7 of module 'lapack' does not match runtime version 3.6
return f(*args, **kwds)
Traceback (most recent call last):
File "", line 1, in
File "/anaconda3/lib/python3.6/site-packages/jax/init.py", line 18, in
import jax.numpy as np # side-effecting import sets up operator overloads
File "/anaconda3/lib/python3.6/site-packages/jax/numpy/init.py", line 18, in
from . import linalg
File "/anaconda3/lib/python3.6/site-packages/jax/numpy/linalg.py", line 24, in
from .. import lax_linalg
File "/anaconda3/lib/python3.6/site-packages/jax/lax_linalg.py", line 32, in
from jaxlib import lapack
File "jaxlib/lapack.pyx", line 77, in init lapack
File "jaxlib/lapack.pyx", line 42, in lapack.register_cpu_custom_call_target
File "/anaconda3/lib/python3.6/site-packages/jaxlib/xla_client.py", line 1574, in register_cpu_custom_call_target
c_api.RegisterCpuCustomCallTarget(name, fn)
UnboundLocalError: local variable 'name' referenced before assignment