jax

0d99f31f - Backport cuda_array_interface , testDotAlgorithm, Optimizer test fixes from rocm-jaxlib-v0.8.0 (#635)