jax
9dbc4119 - GESVDJ support for ROCm GPUs in JAX (#580)

Commit

32 days ago

GESVDJ support for ROCm GPUs in JAX (#580) FFI support for the GESVDJ call is added to support the SVD unit tests.

References

#580 - GESVDJ support for ROCm GPUs in JAX

#599 - Use xla from rocm-jaxlib-v0.8.0 release branch

#670 - Test pytest abort plugin

#701 - Skip test_tridiagonal_solve_grad test 0.8.0

#708 - Skip test_batch_axis_sharding_jvp13 test 0.8.0

Author

tsrw2048

tsrw2048

Parents

Loading