jax
ed8596b0 - Fix GPU lowering rule for SVD on ROCm devices (#600)

Commit

186 days ago

Fix GPU lowering rule for SVD on ROCm devices (#600)

References

#601 - porting changes from 0.8.0 to 0.8.2

#632 - Add script for running bazel tests on ROCm

#638 - Support wheels as build deps

#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

tsrw2048

tsrw2048

Committer

Ruturaj4

Ruturaj4

Parents

Loading