jax

b7ec636c - Merge pull request #4616 from bchetioui:test_linear_solve