jax
84c7adfd - Backport cuda_array_interface , testDotAlgorithm, Optimizer test fixes from rocm-jaxlib-v0.8.0 (#635)

Commit

267 days ago

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

References

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

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

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

#786 - Cherry-pick ROCm fixes from PR #740 to rocm-jaxlib-v0.8.2

Author

magaonka-amd

magaonka-amd

Parents

Loading