google / jax

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

Home Page:http://jax.readthedocs.io/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

scipy.linalg.tril and .triu changed to sparse

rubimat opened this issue · comments

Description

When using jax._src.scipy.linalg I get an error saying:

AttributeError: module 'scipy.linalg' has no attribute 'tril'
AttributeError: module 'scipy.linalg' has no attribute 'triu'

It seems that scipy changed the function namespaces to scipy.sparse.tril and scipy.sparse.triu.

System info (python version, jaxlib version, accelerator, etc.)

jax:    0.4.17
jaxlib: 0.4.17
numpy:  1.26.4
python: 3.11.3 (v3.11.3:f3909b8bc8, Apr  4 2023, 20:12:10) [Clang 13.0.0 (clang-1300.0.29.30)]
jax.devices (1 total, 1 local): [CpuDevice(id=0)]
process_count: 1

Resolved when installing latest release 0.4.16.