Peter Hawkins
|
ba557d5e1b
|
Change JAX's copyright attribution from "Google LLC" to "The JAX Authors.".
See https://opensource.google/documentation/reference/releasing/contributions#copyright for more details.
PiperOrigin-RevId: 476167538
|
2022-09-22 12:27:19 -07:00 |
|
Jake VanderPlas
|
4f62cef1f5
|
[x64] Make TPU svd compatible with strict type promotion
|
2022-06-08 15:38:27 -07:00 |
|
Peter Hawkins
|
db73670ec3
|
Add support for padded arrays in QDWH algorithm.
This change is in preparation for adding a jit-table QDWH-eig implementation.
PiperOrigin-RevId: 448571523
|
2022-05-13 13:57:36 -07:00 |
|
Tianjian Lu
|
4bc1c1c004
|
[linalg] Add svd on zero matrix.
PiperOrigin-RevId: 447521398
|
2022-05-09 11:29:22 -07:00 |
|
Tianjian Lu
|
1093559856
|
[linalg] Add matmul precision scope for svd.
PiperOrigin-RevId: 447095391
|
2022-05-06 16:33:50 -07:00 |
|
Tianjian Lu
|
020849076c
|
[linalg] Add tpu svd lowering rule.
PiperOrigin-RevId: 445533767
|
2022-04-29 16:43:53 -07:00 |
|
Tianjian Lu
|
455c9f823e
|
[linalg] Adds full_matrices option to TPU SVD.
PiperOrigin-RevId: 443163571
|
2022-04-20 12:32:00 -07:00 |
|
Tianjian Lu
|
5a1c5ba114
|
[linalg] Adds compute_uv to TPU SVD.
PiperOrigin-RevId: 442864883
|
2022-04-19 11:28:43 -07:00 |
|
Tianjian Lu
|
5a012d5e7b
|
[JAX] Added jit-able singular value decomposition.
PiperOrigin-RevId: 426193395
|
2022-02-03 11:16:55 -08:00 |
|