jax
164fc708 - Phambinh/skip fused attention navi lds limit (#720)

Commit

38 days ago

Phambinh/skip fused attention navi lds limit (#720)

References

rocm-jaxlib-v0.8.2

#720 - Phambinh/skip fused attention navi lds limit

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

#742 - Fix deprecated mpmath.ctx_mp.mpnumeric import for mpmath >= 1.4.0

Author

phambinhfin

phambinhfin

Parents

Loading