GitHub

linalg.solve_triangular doesn't scale well in parallel for scipy>1.4.1

For very specific uses of solve_triangular, the parallelization performance seems to degrade after 1.4.1. Here is an short example: # Imports from multiprocessing import Pool import time import numpy ...
GitHub

jax.scipy.linalg.solve_triangular

When dim>256, mat result is wrong. mat[j]=DeviceArray([[1., 1., 1., ..., 1., 1., 1.], [1., 1., 1., ..., 1., 1., 1.], [1., 1., 1., ..., 1., 1., 1.], ..., [1., 1., 1 ...