You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Is there any way to use sparse direct solvers in JAX? I can use ‘scipy’ to compute sparse factorizations and convert them to sparse JAX tensors, but then I am stuck because I cannot find any sparse triangular solver. Is anybody aware of a way to circumvent this, in a way that can be jit-compiled?
reacted with thumbs up emoji reacted with thumbs down emoji reacted with laugh emoji reacted with hooray emoji reacted with confused emoji reacted with heart emoji reacted with rocket emoji reacted with eyes emoji
-
Is there any way to use sparse direct solvers in JAX? I can use ‘scipy’ to compute sparse factorizations and convert them to sparse JAX tensors, but then I am stuck because I cannot find any sparse triangular solver. Is anybody aware of a way to circumvent this, in a way that can be jit-compiled?
Beta Was this translation helpful? Give feedback.
All reactions