jax
694e630d - Added changes from PR #626 and PR #645. This also fixes merge conflicts from upstream.

Commit

21 days ago

Added changes from PR #626 and PR #645. This also fixes merge conflicts from upstream. This uses the following commits: - 61625381cc2f014972e9a67692f3249e6ca473d9 - 9344b2b2a0827236b1d11feb63a7c6dbc4286124

References

#650 - Port commits for UT fixes from rocm-jaxlib-v0.8.0 to rocm-jaxlib-v0.8.2: condition number, rnn workspace, and Pallas Triton tests

#704 - Skip test_tridiagonal_solve_grad test 0.8.2

#706 - Backporting `device test` unit tests for ROCm from JAX v0.9.0 to v0.8.2.

#707 - Skip test_batch_axis_sharding_jvp13 test on 0.8.2

#710 - fix poor performance of slogdet when input dimension n=1 or 2

Author

tsrw2048

Committer

Ruturaj4

Parents

5b3e12a7

jax 694e630d - Added changes from PR #626 and PR #645. This also fixes merge conflicts from upstream.

jax
694e630d - Added changes from PR #626 and PR #645. This also fixes merge conflicts from upstream.