mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
11,530 Commits 506 Branches 263 Tags
Commit Graph

5 Commits

Author SHA1 Message Date
Tianjian Lu
d57e36416f [linalg] Update qdwh to prevent underflow in norm estimation. 
PiperOrigin-RevId: 446887070
 2022-05-05 20:12:32 -07:00
Tianjian Lu
2d17b4d637 [linalg] Fix type promotion in QDWH. 
PiperOrigin-RevId: 432352839
 2022-03-03 20:59:28 -08:00
Tianjian Lu
19554e21d3 Enable QDWH TPU tests. 2021-11-30 15:47:50 -08:00
Tianjian Lu
21fa3b3f3b Change QR import in lax.linalg.qdwh. 2021-11-15 22:36:27 -08:00
Tianjian Lu
c5f73b3d8e [JAX] Added jax.lax.linalg.qdwh. 
PiperOrigin-RevId: 406453671
 2021-10-29 14:45:06 -07:00