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
|
e888e7c10c
|
[x64] make lax_scipy_test.py compatible with strict dtype promotion
|
2022-06-14 10:02:45 -07:00 |
|
Jake VanderPlas
|
4f62cef1f5
|
[x64] Make TPU svd compatible with strict type promotion
|
2022-06-08 15:38:27 -07:00 |
|
Jeppe Klitgaard
|
17de89b16a
|
feat: refactor code using pyupgrade
This PR upgrades legacy Python code to 3.7+ code using pyupgrade:
```sh
pyupgrade --py37-plus --keep-runtime-typing **.py
```
a
|
2022-05-17 22:14:05 +01: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 |
|
Peter Hawkins
|
7ba36fc178
|
Change implementation of jax.scipy.linalg.polar() and jax._src.scipy.eigh to use the QDWH decomposition from jax._src.lax.qdwh.
Remove jax._src.lax.polar.
PiperOrigin-RevId: 448241206
|
2022-05-12 07:20:52 -07:00 |
|
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 |
|