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

4 Commits

Author SHA1 Message Date
Tianjian Lu
c5f73b3d8e [JAX] Added jax.lax.linalg.qdwh. 
PiperOrigin-RevId: 406453671
 2021-10-29 14:45:06 -07:00
Leello Tadesse Dadi
f9a246ac19 schur lapack wrapper 2021-09-29 14:29:52 +02:00
Tom Hennigan
ffac40a2c0 Add lax.linalg.tridiagonal_solve(..), lowering to cusparse_gtsv2<T>() on GPU. 
Fixes #6830.
 2021-06-02 13:49:02 +00:00
Peter Hawkins
575a8e0668 Move lax linear algebra routines into a jax.lax.linalg module. 
PiperOrigin-RevId: 340717634
 2020-11-04 13:36:28 -08:00