scipy.linalg.tril and .triu changed to sparse
rubimat opened this issue · comments
rubimat commented
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
rubimat commented
Resolved when installing latest release 0.4.16.