jax

1815f17a - Be more defensive about old jaxlib versions and non-CPU devices in LU decomposition usage.