Tianjian Lu

icon indicating a website link https://scholar.google.com/citations?user=eWEj9g0AAAAJ&hl=en

icon company Google LLC icon location Mountain View, CA icon location Software Engineer at Google Research. Interests in high performance computing, multiphysics simulations, numerical linear algebra, and hardware accelerators.

Results 13 comments of Tianjian Lu

jax.scipy.sparse.linalg.cg inconsistent results between runs

Thanks @fbusato ! Can you share more information on the versions in 11.x when the four algorithms become available? It seems CUSPARSE_SPMM_CSR_ALG3 is in since 11.2 and the other three...

jax.scipy.sparse.linalg.cg inconsistent results between runs

Thanks! It works like a charm. need some patience though :)

jnp.linalg.norm() underflows

@tlu7