jax
498f7350 - Enable lax backend scipy tests on ROCm GPUs (#687)

Commit

76 days ago

Enable lax backend scipy tests on ROCm GPUs (#687)

References

#671 - Rocm jaxlib v0.9.0

#687 - Enable lax backend scipy tests on ROCm GPUs

#699 - Skip testEighTinyNorm test 0.9.0

#725 - Skip test_pmap on ROCm due to IndivisibleError with new pmap SPMD tiling

#730 - Revert "Fix HIP memory leaks in RNN kernels"

#732 - update test

#734 - Add ROCm gesdd SVD backend, export back-compat data, and relaxed GPU …

#739 - Fix: Scaled Matmul at rocm 9.0

Author

magaonka-amd

magaonka-amd

Parents

Loading