Kenneth Meyer

Different Results on GPU and CPU for BiCGSTAB linear solve

4

Hello, I'm using lineax as a linear solver backend for some JAX-based Finite Element code I'm developing (to the lineax developers, thank you! lineax, equinox, etc are amazing IMO), and...

### Is your feature request related to a problem? Please describe. Hello, I raised an issue on the JAX Github that attempted to improve the numerical stability of BiCGStab (jax-ml/jax#32978)...