| a683186 | Yash Katariya | 15 November 2022, 04:37:11 UTC | Use the 11/09 libtpu build for jaxlib release since that passes all the tests. PiperOrigin-RevId: 488543322 | 15 November 2022, 04:37:41 UTC |
| 7635df8 | Parker Schuh | 15 November 2022, 02:44:53 UTC | Remove custom potrf kernels in favor of native XLA cholesky support. PiperOrigin-RevId: 488525158 | 15 November 2022, 02:45:25 UTC |
| f36084a | Yash Katariya | 15 November 2022, 02:30:36 UTC | Update the values for jaxlib release (again) PiperOrigin-RevId: 488522992 | 15 November 2022, 02:31:08 UTC |
| 0da02dd | Yash Katariya | 15 November 2022, 01:08:30 UTC | Update the values needed for a jaxlib release PiperOrigin-RevId: 488508360 | 15 November 2022, 01:08:59 UTC |
| 465d4ec | jax authors | 15 November 2022, 00:58:33 UTC | Merge pull request #13243 from google:yashk2810-patch-19 PiperOrigin-RevId: 488506173 | 15 November 2022, 00:58:33 UTC |
| 50f7512 | Yash Katariya | 15 November 2022, 00:37:09 UTC | Update WORKSPACE for jaxlib release | 15 November 2022, 00:37:09 UTC |
| e6c4d4a | Yash Katariya | 14 November 2022, 23:47:06 UTC | Add docstrings for `Sharding` classes. Right now I am only documenting `Sharding`, `XLACompatibleSharding`, `MeshPspecSharding` and `SingleDeviceSharding`. Also moving jax_array_migration guide to reference documentation. PiperOrigin-RevId: 488489503 | 14 November 2022, 23:47:46 UTC |
| c42bad8 | Yash Katariya | 14 November 2022, 22:43:26 UTC | Make `MeshPspecSharding` an alias for `NamedSharding` (it was the other way around before this CL). PiperOrigin-RevId: 488473538 | 14 November 2022, 22:44:00 UTC |
| d1e26d9 | jax authors | 14 November 2022, 21:48:28 UTC | Merge pull request #13139 from mattjj:djax-vmap4 PiperOrigin-RevId: 488458141 | 14 November 2022, 21:48:28 UTC |
| f715ca6 | jax authors | 14 November 2022, 20:06:04 UTC | Merge pull request #13236 from skye:build_chat PiperOrigin-RevId: 488432432 | 14 November 2022, 20:06:04 UTC |
| 5da7976 | Skye Wanderman-Milne | 14 November 2022, 19:44:45 UTC | Send message to internal chat room on Cloud TPU CI failure | 14 November 2022, 19:44:45 UTC |
| da130cb | Peter Hawkins | 14 November 2022, 18:33:53 UTC | Disable more tests under tsan/asan. PiperOrigin-RevId: 488406459 | 14 November 2022, 18:34:30 UTC |
| 7c6a65b | John QiangZhang | 14 November 2022, 18:11:29 UTC | Add tf.get_current_name_scope() as prefix of name_stack during tf execution. PiperOrigin-RevId: 488399560 | 14 November 2022, 18:12:04 UTC |
| b086e73 | jax authors | 14 November 2022, 17:10:39 UTC | Merge pull request #13189 from Ishticode:lcm_update PiperOrigin-RevId: 488383042 | 14 November 2022, 17:10:39 UTC |
| 4f15563 | jax authors | 14 November 2022, 16:11:36 UTC | Merge pull request #13234 from hawkinsp:docs PiperOrigin-RevId: 488369070 | 14 November 2022, 16:11:36 UTC |
| ce17ce0 | Peter Hawkins | 14 November 2022, 15:35:41 UTC | Mention in the pmap() documentation that all devices must be identical. Fixes https://github.com/google/jax/issues/13203 | 14 November 2022, 15:43:53 UTC |
| ebd9840 | Peter Hawkins | 14 November 2022, 15:38:39 UTC | Add several recent changes to the CHANGELOG. PiperOrigin-RevId: 488362198 | 14 November 2022, 15:39:13 UTC |
| 7600cc8 | Yash Katariya | 14 November 2022, 15:27:24 UTC | Make jax.Array default to False for external colab. PiperOrigin-RevId: 488360010 | 14 November 2022, 15:28:00 UTC |
| aa658bd | Peter Hawkins | 14 November 2022, 15:11:26 UTC | Disable asan/tsan for a number of slow tests. PiperOrigin-RevId: 488356786 | 14 November 2022, 15:12:16 UTC |
| 40e81c3 | Peter Hawkins | 12 November 2022, 00:23:31 UTC | Revert: Use pinv to compute lstsq. The current implementation of lstsq is equivalent to pinv(A) @ b, with a different order of matrix multiplications. If we write it that way we benefit from a more stable derivative that does not require differentiating through the singular value decomposition. This PR appears to have caused numerical problems in downstream tests. PiperOrigin-RevId: 487942754 | 12 November 2022, 00:24:05 UTC |
| d4cc588 | jax authors | 12 November 2022, 00:11:27 UTC | Merge pull request #13215 from ROCmSoftwarePlatform:sytrd PiperOrigin-RevId: 487940256 | 12 November 2022, 00:11:27 UTC |
| e15619c | Sharad Vikram | 11 November 2022, 23:23:44 UTC | Convert string axis name into tuple of strings in Mesh constructor PiperOrigin-RevId: 487930412 | 11 November 2022, 23:27:51 UTC |
| 6897d37 | Yash Katariya | 11 November 2022, 23:20:27 UTC | Add docstrings for jax.Array APIs `make_array_from_callback` and `make_array_from_single_device_arrays`. PiperOrigin-RevId: 487929688 | 11 November 2022, 23:21:10 UTC |
| 31d8f62 | Rahul Batra | 11 November 2022, 22:41:51 UTC | Sytrd solver and SytrdDescriptor should NOT be CUDA only | 11 November 2022, 22:41:51 UTC |
| 09f62de | Ishtiaq Hussain | 10 November 2022, 19:03:04 UTC | Moved abs to inputs of lcm and added specific test | 11 November 2022, 22:31:06 UTC |
| 19d76a7 | jax authors | 11 November 2022, 22:18:19 UTC | Merge pull request #13212 from sharadmv:fix-changelog PiperOrigin-RevId: 487915320 | 11 November 2022, 22:18:19 UTC |
| 4bdfdd7 | Sharad Vikram | 11 November 2022, 22:00:33 UTC | Update changelog w/ info about deleting `jax_experimental_name_stack` | 11 November 2022, 22:02:30 UTC |
| 7c3fb81 | Peter Hawkins | 11 November 2022, 21:28:11 UTC | Use pinv to compute lstsq. The current implementation of lstsq is equivalent to pinv(A) @ b, with a different order of matrix multiplications. If we write it that way we benefit from a more stable derivative that does not require differentiating through the singular value decomposition. PiperOrigin-RevId: 487903227 | 11 November 2022, 21:28:48 UTC |
| 047974d | Peter Hawkins | 11 November 2022, 20:55:20 UTC | Be more economical when computing the JVP of the SVD of non-square matrices. (Note this isn't a regression from #13147: the previous change did not alter the order of operations.) PiperOrigin-RevId: 487896154 | 11 November 2022, 20:55:53 UTC |
| c9ebf60 | Peter Hawkins | 11 November 2022, 19:44:22 UTC | Compute the JVP of jnp.linalg.pinv more economically for non-square matrices. The order of the matrix products matters. PiperOrigin-RevId: 487879202 | 11 November 2022, 19:45:02 UTC |
| 9957361 | jax authors | 11 November 2022, 18:31:24 UTC | Merge pull request #13198 from patrick-kidger:prng-isinstance PiperOrigin-RevId: 487859092 | 11 November 2022, 18:31:24 UTC |
| c359c79 | jax authors | 11 November 2022, 17:01:55 UTC | Merge pull request #13196 from jakevdp:simpler-sparsify PiperOrigin-RevId: 487837115 | 11 November 2022, 17:01:55 UTC |
| d2afa84 | Patrick Kidger | 11 November 2022, 16:04:38 UTC | PRNGKeyArray is now a virtual subclass of ndarray | 11 November 2022, 16:04:38 UTC |
| 10e6fe8 | Felix Chern | 11 November 2022, 15:08:27 UTC | Correct norm in ann.py doc. PiperOrigin-RevId: 487814084 | 11 November 2022, 15:08:54 UTC |
| ce85106 | jax authors | 11 November 2022, 14:51:30 UTC | Merge pull request #13193 from tlu7:bcsr-fromdense-batching PiperOrigin-RevId: 487810600 | 11 November 2022, 14:51:30 UTC |
| 71f92a7 | jax authors | 11 November 2022, 14:44:29 UTC | Merge pull request #13182 from canyon289:docs_update PiperOrigin-RevId: 487808462 | 11 November 2022, 14:44:29 UTC |
| bdf3bd5 | jax authors | 11 November 2022, 14:37:30 UTC | Merge pull request #13155 from jakevdp:bcoo-gather PiperOrigin-RevId: 487808456 | 11 November 2022, 14:37:30 UTC |
| ea25b79 | Jake VanderPlas | 11 November 2022, 12:50:33 UTC | [sparse] streamline sparse rules for standard primitives | 11 November 2022, 12:50:33 UTC |
| 90dc008 | Jake VanderPlas | 11 November 2022, 12:25:14 UTC | [sparse] add bcoo_gather & support for sparse indexing | 11 November 2022, 12:25:14 UTC |
| a135414 | Peter Hawkins | 11 November 2022, 03:37:26 UTC | Reenable a TPU test now that the compiler bug is fixed. PiperOrigin-RevId: 487705048 | 11 November 2022, 03:38:01 UTC |
| 332fced | Tianjian Lu | 10 November 2022, 21:55:35 UTC | sparse] BCSR batching rule. [Co-authored-by: Jake Vanderplas: <vanderplas@google.com> | 11 November 2022, 03:33:32 UTC |
| dc0d7ba | jax authors | 11 November 2022, 03:15:29 UTC | Merge pull request #13202 from google:yashk2810-patch-18 PiperOrigin-RevId: 487701933 | 11 November 2022, 03:15:29 UTC |
| f0c0689 | Yash Katariya | 11 November 2022, 03:09:19 UTC | Remove internal information | 11 November 2022, 03:09:19 UTC |
| 4a3b7f1 | Parker Schuh | 11 November 2022, 03:04:32 UTC | Change pickling for jax.sharding to not serialize device ids. PiperOrigin-RevId: 487700467 | 11 November 2022, 03:05:02 UTC |
| f9d7a6a | jax authors | 11 November 2022, 01:56:29 UTC | Merge pull request #13197 from google:yashk2810-patch-17 PiperOrigin-RevId: 487687006 | 11 November 2022, 01:56:29 UTC |
| 73935a5 | Yash Katariya | 11 November 2022, 01:23:16 UTC | Update jax_array_migration.md | 11 November 2022, 01:23:16 UTC |
| c318f77 | jax authors | 11 November 2022, 01:18:44 UTC | Merge pull request #13185 from tlu7:bcsr-from-scipy PiperOrigin-RevId: 487680265 | 11 November 2022, 01:18:44 UTC |
| aa66b93 | Yash Katariya | 11 November 2022, 01:08:57 UTC | Fix the docs build | 11 November 2022, 01:08:57 UTC |
| b49a1bd | Yash Katariya | 11 November 2022, 00:45:51 UTC | Add jax.Array migration doc to OSS PiperOrigin-RevId: 487673643 | 11 November 2022, 00:46:30 UTC |
| 311fb24 | Tianjian Lu | 10 November 2022, 07:01:24 UTC | [sparse] Add BCSR from_scipy_sparse. Co-authored-by: Jake Vanderplas <vanderplas@google.com> | 11 November 2022, 00:44:59 UTC |
| 352b042 | Peter Hawkins | 10 November 2022, 21:15:44 UTC | Add a GPU implementation of symmetric (Hermitian) tridiagonal reduction. Change the contract of lax.linalg.tridiagonal to return the d and e vectors as well. Since we only just added this function and have never released JAX with it we can make this change without breaking compatibility. Also fix wrong dtypes for d and e values in the CPU lapack sytrd wrapper. PiperOrigin-RevId: 487621469 | 10 November 2022, 21:16:21 UTC |
| 74b136e | Sharad Vikram | 10 November 2022, 19:59:16 UTC | Delete `jax_experimental_name_stack` flag PiperOrigin-RevId: 487601864 | 10 November 2022, 19:59:50 UTC |
| 0ebb6b4 | jax authors | 10 November 2022, 18:04:35 UTC | Merge pull request #13180 from jakevdp:bcoo-slice PiperOrigin-RevId: 487568853 | 10 November 2022, 18:04:35 UTC |
| cc41ee8 | Yash Katariya | 10 November 2022, 15:38:21 UTC | Mark scipy_signal_test and sparse_test `optonly` because it times out under debug mode. PiperOrigin-RevId: 487533356 | 10 November 2022, 15:38:58 UTC |
| 71360ed | Yash Katariya | 10 November 2022, 04:31:32 UTC | Bump the shard count for TPU to avoid timeouts PiperOrigin-RevId: 487421018 | 10 November 2022, 04:32:12 UTC |
| e42e52d | Peter Hawkins | 10 November 2022, 02:57:28 UTC | Rename test flag --num_generated_cases to --jax_num_generated_cases. parse_flags_with_absl() only parses flags that start with --jax_. Other flags are only parsed when absl.app's main function runs. But that's too late for test cases: test cases need to have the number of generated cases chosen at module initialization time. Hence the --num_generated_cases flag wasn't doing anything. Oops. By renaming it it works once again. It might make sense to stop using flags for the number of generated cases and only use environment variables. We defer that to a future change. Fix many test cases that were shown to be broken with a larger number of test cases enabled. PiperOrigin-RevId: 487406670 | 10 November 2022, 02:58:05 UTC |
| b36afc5 | jax authors | 10 November 2022, 01:29:30 UTC | Merge pull request #13177 from jakevdp:bcoo-dynamic-slice PiperOrigin-RevId: 487390430 | 10 November 2022, 01:29:30 UTC |
| 3731e44 | Sharad Vikram | 10 November 2022, 01:18:19 UTC | Set default layout for Python callback PiperOrigin-RevId: 487388682 | 10 November 2022, 01:18:49 UTC |
| f9bbd58 | Yash Katariya | 10 November 2022, 00:37:29 UTC | Improve the error message when `@pjit` (with no {in_axis|out_axis}_resources is used without jax.Array enabled. PiperOrigin-RevId: 487380328 | 10 November 2022, 00:38:00 UTC |
| cb3762e | Ravin Kumar | 08 November 2022, 22:45:58 UTC | Consolidate links in JAX documentation Move notes down | 09 November 2022, 23:44:52 UTC |
| 4c4f2a3 | Jake VanderPlas | 09 November 2022, 23:03:21 UTC | [sparse] support strides in bcoo_slice | 09 November 2022, 23:03:21 UTC |
| b41c594 | John QiangZhang | 09 November 2022, 22:40:59 UTC | Internal change PiperOrigin-RevId: 487351591 | 09 November 2022, 22:41:39 UTC |
| 46d9cac | Jake VanderPlas | 09 November 2022, 21:26:27 UTC | [sparse] bcoo_dynamic_slice: remove unnecessary padding from output | 09 November 2022, 21:56:18 UTC |
| fa0217b | jax authors | 09 November 2022, 21:34:55 UTC | Merge pull request #13175 from jakevdp:bcoo-transpose PiperOrigin-RevId: 487333413 | 09 November 2022, 21:34:55 UTC |
| 0c3e330 | Jake VanderPlas | 09 November 2022, 20:53:13 UTC | [sparse] fix shape bug in bcoo_transpose | 09 November 2022, 20:53:13 UTC |
| 8ac7422 | Felix Chern | 09 November 2022, 19:32:12 UTC | [JAX] Disables large k test cases in ann_test. Will investigate probability properties for the corner cases in the future. PiperOrigin-RevId: 487302143 | 09 November 2022, 19:32:47 UTC |
| 63e3152 | jax authors | 09 November 2022, 18:18:22 UTC | Merge pull request #13160 from jakevdp:bcoo-squeeze PiperOrigin-RevId: 487280563 | 09 November 2022, 18:18:22 UTC |
| f697b8e | jax authors | 09 November 2022, 17:31:39 UTC | Merge pull request #13166 from LenaMartens:checking-keys PiperOrigin-RevId: 487267267 | 09 November 2022, 17:31:39 UTC |
| 053b8b5 | lenamartens | 09 November 2022, 12:08:57 UTC | Checkify: fix nan_checks+PRNGKeys - a PRNGKey is never NaN! Add a guard to the nan_error_rule to not call jnp.isnan on keys. | 09 November 2022, 17:08:21 UTC |
| 1cead77 | Peter Hawkins | 09 November 2022, 14:23:22 UTC | Add support for Hessenberg and tridiagonal matrix reductions on CPU. * Implement jax.scipy.linalg.hessenberg and jax.lax.linalg.hessenberg. * Export what was previously jax._src.lax.linalg.orgqr as jax.lax.linalg.householder_product, since it can be used with some minor tweaks to compute the unitary matrix of a Hessenberg reduction. * Implement jax.lax.linalg.tridiagonal, which is the symmetric (Hermitian) equivalent of Hessenberg reduction. None of these primitives are differentiable at the moment. PiperOrigin-RevId: 487224934 | 09 November 2022, 14:23:55 UTC |
| 30637d0 | jax authors | 09 November 2022, 13:50:34 UTC | Merge pull request #13168 from hawkinsp:fixbuild PiperOrigin-RevId: 487219149 | 09 November 2022, 13:50:34 UTC |
| 41c90a8 | Peter Hawkins | 09 November 2022, 13:37:49 UTC | Add missing stablehlo dialect files to jaxlib build. Unbreaks the build. | 09 November 2022, 13:37:49 UTC |
| 3b1ddf2 | Tianjian Lu | 09 November 2022, 07:02:39 UTC | [linalg] Add jax.scipy.special.bessel_jn (Bessel function of the first kind). PiperOrigin-RevId: 487146250 | 09 November 2022, 07:03:21 UTC |
| 5599632 | Eugene Burmako | 09 November 2022, 06:49:26 UTC | Introduce XlaLowering::stablehlo() and use it in associated APIs See tests/api_test.py for usage examples. At the moment, stablehlo() works by using the hlo-legalize-to-stablehlo pass, which takes MHLO natively produced by JAX and converts it into StableHLO. This is an intermediate step towards switching JAX to natively produce StableHLO. This CL adds both mhlo_to_stablehlo and stablehlo_to_mhlo to jaxlib, even though only the former is used at the moment. This is done in anticipation of switching JAX to natively produce StableHLO, where stablehlo_to_mhlo will be needed to provide backward compatibility for XlaLowering::mhlo(). We're adding stablehlo_to_mhlo now, so that in the future we don't have to update jaxlib again which will make deployment easier. PiperOrigin-RevId: 487144342 | 09 November 2022, 06:50:06 UTC |
| df963bd | Skye Wanderman-Milne | 09 November 2022, 04:05:59 UTC | Remove flaky Array defragmentation test check PiperOrigin-RevId: 487120630 | 09 November 2022, 04:06:36 UTC |
| 0cf220f | jax authors | 09 November 2022, 02:37:45 UTC | Merge pull request #13162 from jakevdp:bcoo-reshape PiperOrigin-RevId: 487106270 | 09 November 2022, 02:37:45 UTC |
| 4255697 | Jake VanderPlas | 09 November 2022, 02:16:20 UTC | [sparse] add bcoo_squeeze function | 09 November 2022, 02:16:20 UTC |
| 53344b8 | Yash Katariya | 09 November 2022, 01:01:49 UTC | Don't create copies by device_putting a host local jax.Array if the sharding matches with the input. PiperOrigin-RevId: 487090094 | 09 November 2022, 01:02:23 UTC |
| 7d3b1d6 | Jake VanderPlas | 09 November 2022, 01:00:25 UTC | [sparse] fix bcoo_reshape under jit | 09 November 2022, 01:00:25 UTC |
| 0d2cd6d | Skye Wanderman-Milne | 08 November 2022, 23:31:53 UTC | [jax] Fix manual defragment method to work with Arrays PiperOrigin-RevId: 487068409 | 08 November 2022, 23:32:30 UTC |
| 5e1d7cd | jax authors | 08 November 2022, 23:01:23 UTC | Merge pull request #13032 from jakevdp:sharding-attr PiperOrigin-RevId: 487061046 | 08 November 2022, 23:01:23 UTC |
| 8fbf8da | Jake VanderPlas | 08 November 2022, 22:20:46 UTC | Declare Array.sharding & raise an error on tracers | 08 November 2022, 22:20:46 UTC |
| af017d4 | jax authors | 08 November 2022, 22:11:51 UTC | Merge pull request #13153 from jakevdp:bcoo-reshape PiperOrigin-RevId: 487046508 | 08 November 2022, 22:11:51 UTC |
| 768076e | jax authors | 08 November 2022, 22:05:09 UTC | Merge pull request #13157 from jakevdp:bcoo-astype PiperOrigin-RevId: 487046458 | 08 November 2022, 22:05:09 UTC |
| 7c0d0e6 | Jake VanderPlas | 08 November 2022, 21:30:22 UTC | [sparse] add support for BCOO.astype method | 08 November 2022, 21:30:22 UTC |
| af95663 | Jake VanderPlas | 08 November 2022, 20:00:24 UTC | [sparse] fix bcoo_reshape when n_sparse=0 | 08 November 2022, 20:00:24 UTC |
| 96f6c1c | Yuxin Wu | 08 November 2022, 18:49:22 UTC | Let is_user_frame ignore frames from stdlib. When using decorators, we found contextlib.py from stdlib sometimes become the most recent non-jax frame. But it's not a user frame. PiperOrigin-RevId: 486993924 | 08 November 2022, 18:50:08 UTC |
| 0b463ef | Matthew Johnson | 07 November 2022, 06:56:51 UTC | tighten up vmap w/ piles: require pile_axis in_axes/out_axes | 08 November 2022, 18:27:55 UTC |
| 500cd85 | jax authors | 08 November 2022, 17:57:44 UTC | Merge pull request #13144 from LenaMartens:donate-no-more PiperOrigin-RevId: 486979733 | 08 November 2022, 17:57:44 UTC |
| 3994ac3 | jax authors | 08 November 2022, 14:39:54 UTC | Merge pull request #13145 from hawkinsp:pinv PiperOrigin-RevId: 486935918 | 08 November 2022, 14:39:54 UTC |
| ab8cde9 | Peter Hawkins | 07 November 2022, 16:39:19 UTC | Add support for the hermitian option on jnp.linalg.pinv. Improve the pinv implementation to avoid computing an unnecessary reduction: svd sorts its singular values so we don't need to use amax() to find the largest one. Avoid explicitly forming the identity matrix in the pinv JVP. | 08 November 2022, 13:53:00 UTC |
| e80c34d | lenamartens | 07 November 2022, 15:46:53 UTC | Don't donate arguments in jit/pmap/pjit when debug_nans=True. | 08 November 2022, 13:33:59 UTC |
| 1e7e8e8 | jax authors | 08 November 2022, 03:25:15 UTC | Merge pull request #13147 from hawkinsp:eyes PiperOrigin-RevId: 486826532 | 08 November 2022, 03:25:15 UTC |
| 85f43dd | jax authors | 08 November 2022, 02:23:41 UTC | Merge pull request #13061 from nouiz:test_doc PiperOrigin-RevId: 486816419 | 08 November 2022, 02:23:41 UTC |
| eb9e8c2 | jax authors | 07 November 2022, 23:31:10 UTC | Merge pull request #13117 from 8bitmp3:move-multihost-multiprocess-toc PiperOrigin-RevId: 486780726 | 07 November 2022, 23:31:10 UTC |
| e00f7e7 | jax authors | 07 November 2022, 21:49:52 UTC | Merge pull request #13093 from PhilipVinc:patch-1 PiperOrigin-RevId: 486753425 | 07 November 2022, 21:49:52 UTC |
| 2c1fe45 | Parker Schuh | 07 November 2022, 21:39:59 UTC | Add UnloadedMeshExecutable to represent a MeshExecutable that is not loaded on any physical devices for the purposes of serialization. This type is easier to serialize because it has not yet been converted into arg-handlers. Potential API: ``` str, in_tree, out_tree = lowered.compile_and_serialize() exec = jax.experimental.load_serialized(str, in_tree, out_tree, backend) exec // identical to lowered.compile(). ``` PiperOrigin-RevId: 486751141 | 07 November 2022, 21:40:40 UTC |
| 793fb9b | Filippo Vicentini | 07 November 2022, 20:28:31 UTC | Fix issue in `check_tree`, so that custom_linear_solve supports hax_aux=True when the vector and the aux are both pytrees. | 07 November 2022, 20:29:43 UTC |
| 2183059 | Hyeontaek Lim | 07 November 2022, 19:54:30 UTC | Fix GDA error message formatting PiperOrigin-RevId: 486724647 | 07 November 2022, 19:55:28 UTC |
| 845f8df | Peter Hawkins | 07 November 2022, 17:40:16 UTC | Avoid forming identity matrix in SVD JVP. Set the default matmul precision in the SVD JVP, and use @ to express matmuls. Also fix a flaky test failure in QR test on Mac ARM. | 07 November 2022, 18:55:45 UTC |
