2dfc5f9 | jax authors | 07 October 2020, 05:59:09 UTC | Merge pull request #4467 from google:update-pypi PiperOrigin-RevId: 335798412 | 07 October 2020, 05:59:09 UTC |
1c5a09a | Matthew Johnson | 07 October 2020, 05:23:29 UTC | update version and changelog for pypi | 07 October 2020, 05:23:29 UTC |
cf1fefc | jax authors | 06 October 2020, 20:33:20 UTC | Merge pull request #4464 from google:minor-source-sync PiperOrigin-RevId: 335709378 | 06 October 2020, 20:33:20 UTC |
5f6fab8 | Roy Frostig | 06 October 2020, 20:24:19 UTC | minor change as source sync test | 06 October 2020, 20:24:19 UTC |
6eb4f54 | jax authors | 06 October 2020, 19:45:43 UTC | Internal change PiperOrigin-RevId: 335699277 | 06 October 2020, 19:53:21 UTC |
93f4dd4 | George Necula | 06 October 2020, 11:51:51 UTC | Revert "[jax2tf] Added Getting Started notebook (#4460)" (#4463) This reverts commit 1d73952736e6142c0670879072cc99327814640e. | 06 October 2020, 11:51:51 UTC |
1d73952 | George Necula | 06 October 2020, 11:14:28 UTC | [jax2tf] Added Getting Started notebook (#4460) * [jax2tf] Added Getting Started notebook * Updated the jax2tf README * Minor cleanup | 06 October 2020, 11:14:28 UTC |
2166f0e | jax authors | 06 October 2020, 00:41:30 UTC | Merge pull request #4457 from google:tycheck-invar-eqn-context PiperOrigin-RevId: 335537630 | 06 October 2020, 00:41:30 UTC |
7efa320 | jax authors | 05 October 2020, 23:14:32 UTC | Merge pull request #4458 from hawkinsp:xla PiperOrigin-RevId: 335521726 | 05 October 2020, 23:14:32 UTC |
49d851d | Peter Hawkins | 05 October 2020, 20:14:32 UTC | Update XLA. | 05 October 2020, 20:14:32 UTC |
e797925 | Roy Frostig | 05 October 2020, 19:29:43 UTC | equation context for undefined var reads in jaxpr typechecker | 05 October 2020, 19:29:43 UTC |
91755e5 | jax authors | 05 October 2020, 19:09:35 UTC | Merge pull request #4456 from jakevdp:small-fix PiperOrigin-RevId: 335469646 | 05 October 2020, 19:09:35 UTC |
9273e8c | jax authors | 05 October 2020, 19:09:21 UTC | Merge pull request #4455 from johnpjf:patch-2 PiperOrigin-RevId: 335461715 | 05 October 2020, 19:09:21 UTC |
efeed6c | jax authors | 05 October 2020, 19:09:05 UTC | Merge pull request #4444 from jakevdp:dtype-check PiperOrigin-RevId: 335442628 | 05 October 2020, 19:09:05 UTC |
7854a23 | Jake VanderPlas | 05 October 2020, 18:50:04 UTC | remove redundant if-else | 05 October 2020, 18:50:04 UTC |
b357005 | johnpjf | 05 October 2020, 17:19:18 UTC | Update Common_Gotchas_in_JAX.ipynb Clarify that the index is clamped to the bounds of the array when accessing out of bounds. | 05 October 2020, 17:19:18 UTC |
d388b62 | jax authors | 05 October 2020, 14:17:42 UTC | Merge pull request #4442 from hawkinsp:docs PiperOrigin-RevId: 335411058 | 05 October 2020, 14:17:42 UTC |
51538bd | jax authors | 05 October 2020, 05:17:06 UTC | Merge pull request #4446 from n2cholas:master PiperOrigin-RevId: 335349341 | 05 October 2020, 05:17:06 UTC |
238b4a1 | Nicholas Vadivelu | 05 October 2020, 04:54:18 UTC | Update examples/differentially_private_sgd.py Co-authored-by: Matthew Johnson <mattjj@google.com> | 05 October 2020, 04:54:18 UTC |
0e92124 | jax authors | 03 October 2020, 10:23:55 UTC | Merge pull request #4443 from jakevdp:concrete PiperOrigin-RevId: 335180298 | 03 October 2020, 10:23:55 UTC |
2ced163 | Nicholas Vadivelu | 03 October 2020, 06:41:23 UTC | fix dp sgd example | 03 October 2020, 06:41:23 UTC |
6d5ace2 | jax authors | 03 October 2020, 01:12:39 UTC | Merge pull request #4445 from google:lax-ad-test-skip PiperOrigin-RevId: 335136639 | 03 October 2020, 01:12:39 UTC |
db4a3ee | Roy Frostig | 03 October 2020, 00:41:08 UTC | skip test that fails due to known xla:cpu bug | 03 October 2020, 00:41:08 UTC |
0328f85 | Jake VanderPlas | 02 October 2020, 22:31:02 UTC | Call check_user_dtype on all user dtypes | 03 October 2020, 00:21:44 UTC |
5cfd0e9 | jax authors | 02 October 2020, 22:56:04 UTC | Merge pull request #4441 from jakevdp:choice PiperOrigin-RevId: 335117255 | 02 October 2020, 22:56:04 UTC |
492889f | Jake VanderPlas | 02 October 2020, 22:08:21 UTC | Improve errors for failed compilations w/ core.concrete_or_error | 02 October 2020, 22:08:21 UTC |
5732428 | Peter Hawkins | 02 October 2020, 20:44:42 UTC | Fix rendering problems in lax.cond docstring. | 02 October 2020, 20:44:42 UTC |
2fccbbf | Jake VanderPlas | 02 October 2020, 20:13:21 UTC | Implement jnp.choose | 02 October 2020, 20:13:21 UTC |
95e3fcd | jax authors | 02 October 2020, 18:17:20 UTC | Merge pull request #4438 from hawkinsp:gelu PiperOrigin-RevId: 335063459 | 02 October 2020, 18:17:20 UTC |
113c806 | jax authors | 02 October 2020, 18:14:03 UTC | Merge pull request #4439 from hawkinsp:selectandscatter PiperOrigin-RevId: 335062792 | 02 October 2020, 18:14:03 UTC |
d25f404 | jax authors | 02 October 2020, 18:00:20 UTC | Merge pull request #4429 from apaszke:all_to_all_vmap_split PiperOrigin-RevId: 335059726 | 02 October 2020, 18:00:20 UTC |
2d74037 | Peter Hawkins | 02 October 2020, 14:56:08 UTC | Change select_and_scatter_add's batching rule to avoid unrolling. Add a test for select_and_scatter_add batching. | 02 October 2020, 16:02:34 UTC |
9b3bbe8 | Peter Hawkins | 02 October 2020, 13:48:07 UTC | Adds an approximate=... keyword argument to jax.nn.gelu to select between the approximate and exact formulations of gelu. Default to the approximate formulation for now. | 02 October 2020, 13:48:07 UTC |
b609040 | jax authors | 02 October 2020, 02:26:55 UTC | Merge pull request #4413 from alexminnaar:master PiperOrigin-RevId: 334946450 | 02 October 2020, 02:26:55 UTC |
9cf6159 | jax authors | 02 October 2020, 01:34:24 UTC | Merge pull request #4434 from google:numpy-dispatch-test-count PiperOrigin-RevId: 334940761 | 02 October 2020, 01:34:24 UTC |
0893b08 | Alex Minnaar | 21 September 2020, 20:59:46 UTC | histogramdd implemenation | 01 October 2020, 22:02:33 UTC |
fd39901 | Roy Frostig | 01 October 2020, 22:00:38 UTC | reduce test-case count of the numpy-dispatch CI check, to match our other x64-mode CI check | 01 October 2020, 22:00:38 UTC |
93fbae6 | jax authors | 01 October 2020, 21:54:52 UTC | Merge pull request #4431 from tomhennigan:changelist/334843453 PiperOrigin-RevId: 334902700 | 01 October 2020, 21:54:52 UTC |
f18b914 | Tom Hennigan | 01 October 2020, 17:26:07 UTC | Remove scalar float0 XLA representation. Some places in the code (e.g. lower_fun in xla.py) we need to go from an XLA shape to a ShapedArray and cannot do this if we drop shape information earlier. | 01 October 2020, 18:02:31 UTC |
9a8ee95 | jax authors | 01 October 2020, 17:39:51 UTC | Merge pull request #4419 from rsepassi:master PiperOrigin-RevId: 334846970 | 01 October 2020, 17:39:51 UTC |
a622220 | jax authors | 01 October 2020, 14:58:43 UTC | Merge pull request #4430 from LenaMartens:changelist/334791342 PiperOrigin-RevId: 334813454 | 01 October 2020, 14:58:43 UTC |
cc0114a | Lena Martens | 01 October 2020, 12:07:33 UTC | Fix dtype behavior with float0s in CustomVJP. | 01 October 2020, 14:17:51 UTC |
69fda9e | jax authors | 01 October 2020, 02:25:00 UTC | Merge pull request #4039 from LenaMartens:changelist/325216264 PiperOrigin-RevId: 334728148 | 01 October 2020, 02:25:00 UTC |
768b0c1 | jax authors | 30 September 2020, 21:44:47 UTC | Merge pull request #4415 from jakevdp:indices-concrete PiperOrigin-RevId: 334683008 | 30 September 2020, 21:44:47 UTC |
0b23a90 | Jake VanderPlas | 28 September 2020, 22:33:06 UTC | Call concrete_or_error on indices() argument | 30 September 2020, 18:44:49 UTC |
47da512 | jax authors | 30 September 2020, 17:23:34 UTC | Merge pull request #4384 from jakevdp:cov-weights PiperOrigin-RevId: 334626974 | 30 September 2020, 17:23:34 UTC |
62f1e83 | jax authors | 30 September 2020, 17:15:55 UTC | Merge pull request #4425 from akihironitta:use-raise-from PiperOrigin-RevId: 334625280 | 30 September 2020, 17:15:55 UTC |
d707ae1 | Akihiro Nitta | 30 September 2020, 15:27:03 UTC | Merge branch 'master' into use-raise-from | 30 September 2020, 15:27:03 UTC |
e61ca91 | Adam Paszke | 22 September 2020, 13:05:08 UTC | Implement split_axis for all_to_all This allows us to use `all_to_all` over a mix of vmapped and pmapped dimensions, which will be useful for `gmap`. | 30 September 2020, 08:25:36 UTC |
fa38f25 | Adam Paszke | 22 September 2020, 11:19:06 UTC | Add support for all_to_all over vmapped axes | 30 September 2020, 08:25:36 UTC |
2d94a87 | jax authors | 29 September 2020, 20:42:18 UTC | Merge pull request #4421 from jakevdp:multi-buf PiperOrigin-RevId: 334452699 | 29 September 2020, 20:42:18 UTC |
6b5a265 | jax authors | 29 September 2020, 19:07:09 UTC | Merge pull request #4414 from n2cholas:master PiperOrigin-RevId: 334432358 | 29 September 2020, 19:07:09 UTC |
40016cc | Jake VanderPlas | 29 September 2020, 18:53:17 UTC | Allow jax objects to be represented by multiple buffers | 29 September 2020, 18:53:17 UTC |
06170da | Akihiro Nitta | 29 September 2020, 16:20:00 UTC | Use `raise from` | 29 September 2020, 16:20:00 UTC |
97592c8 | Ryan Sepassi | 29 September 2020, 14:57:20 UTC | Add --exclude_test_targets to test_util | 29 September 2020, 14:57:20 UTC |
c67454d | jax authors | 29 September 2020, 13:53:17 UTC | Merge pull request #4417 from google:revert-3983-device-put-tuple PiperOrigin-RevId: 334368988 | 29 September 2020, 13:53:17 UTC |
a0e14b0 | Peter Hawkins | 29 September 2020, 13:26:11 UTC | Revert "Allow JAX objects to be represented by multiple buffers" | 29 September 2020, 13:26:11 UTC |
76e9f23 | Nicholas Vadivelu | 29 September 2020, 02:19:47 UTC | update device faq | 29 September 2020, 02:19:47 UTC |
a51a4d9 | Jake VanderPlas | 28 September 2020, 22:34:57 UTC | Add test coverage for jnp.cov aweights & fweights | 28 September 2020, 22:34:57 UTC |
ecad419 | Lena Martens | 24 September 2020, 15:29:57 UTC | Support grad with integer arguments. - Add float0 and set-up at_least_vspace to return float0 values for int/bool primals - Use Zero to wrap float0 tangents so they're correctly ignored in jvp rules - Add float0 handlers to XLA to support jit - Fix convert_element_type and tie_in jvp rules | 28 September 2020, 18:07:04 UTC |
fa11338 | jax authors | 28 September 2020, 17:00:49 UTC | Merge pull request #3983 from jakevdp:device-put-tuple PiperOrigin-RevId: 334180017 | 28 September 2020, 17:00:49 UTC |
c4a08af | Nicholas Vadivelu | 27 September 2020, 00:36:40 UTC | document setting default device and array creation | 27 September 2020, 00:36:40 UTC |
5b3cbc5 | jax authors | 26 September 2020, 00:44:21 UTC | Merge pull request #4342 from google:improve-tracer-error PiperOrigin-RevId: 333841912 | 26 September 2020, 00:44:21 UTC |
23a25da | Matthew Johnson | 25 September 2020, 23:02:32 UTC | fix test | 25 September 2020, 23:02:32 UTC |
aaa5724 | jax authors | 25 September 2020, 22:42:58 UTC | Merge pull request #4412 from jakevdp:axis-concrete PiperOrigin-RevId: 333822998 | 25 September 2020, 22:42:58 UTC |
61fcea9 | Matthew Johnson | 25 September 2020, 22:35:44 UTC | tweak error to mention *flattened* positions | 25 September 2020, 22:35:44 UTC |
f66c2ee | Matthew Johnson | 25 September 2020, 22:13:14 UTC | Merge branch 'master' into improve-tracer-error | 25 September 2020, 22:13:14 UTC |
185590f | Jake VanderPlas | 25 September 2020, 21:18:46 UTC | Use core.concrete_or_error() to improve errors in reductions | 25 September 2020, 21:18:46 UTC |
4d55683 | jax authors | 25 September 2020, 19:35:24 UTC | Merge pull request #4411 from google:omnistaging-doc-fix-links PiperOrigin-RevId: 333787009 | 25 September 2020, 19:35:24 UTC |
be6e85a | Matthew Johnson | 25 September 2020, 19:10:03 UTC | fix links in omnistaging doc (hopefully...) | 25 September 2020, 19:10:03 UTC |
5e19481 | jax authors | 25 September 2020, 18:59:22 UTC | Merge pull request #4410 from google:add-omnistaging-doc PiperOrigin-RevId: 333779252 | 25 September 2020, 18:59:22 UTC |
90b272a | Matthew Johnson | 25 September 2020, 18:50:32 UTC | add author line, fix typo | 25 September 2020, 18:50:32 UTC |
0fb12a2 | Matthew Johnson | 25 September 2020, 18:48:49 UTC | add omnistaging doc | 25 September 2020, 18:48:49 UTC |
d1f8022 | Jake VanderPlas | 25 September 2020, 18:09:08 UTC | Allow jax objects to be represented by multiple buffers | 25 September 2020, 18:09:08 UTC |
8d24d0a | jax authors | 25 September 2020, 17:56:00 UTC | Merge pull request #4368 from jakevdp:unop-check-array PiperOrigin-RevId: 333765605 | 25 September 2020, 17:56:00 UTC |
95389e8 | jax authors | 25 September 2020, 14:12:58 UTC | Merge pull request #4406 from gnecula:print_debug PiperOrigin-RevId: 333727118 | 25 September 2020, 14:12:58 UTC |
f46a422 | George Necula | 25 September 2020, 12:28:23 UTC | [host_callback] improved logging, and add instructions for debugging | 25 September 2020, 12:28:23 UTC |
ca32966 | jax authors | 25 September 2020, 07:11:36 UTC | Merge pull request #4403 from SIben:eigh-edge-case PiperOrigin-RevId: 333679325 | 25 September 2020, 07:11:36 UTC |
a1da363 | jax authors | 25 September 2020, 06:47:00 UTC | Merge pull request #4402 from gnecula:omni_print PiperOrigin-RevId: 333676897 | 25 September 2020, 06:47:00 UTC |
fd5294b | Benjamin Chetioui | 24 September 2020, 14:46:08 UTC | [jax2tf] Fix casting in translation of eigh for empty arrays. In the case when the operand is complex, we still need to cast the resulting empty eigenvalues to the corresponding floating point type. | 24 September 2020, 14:49:06 UTC |
1bf8036 | jax authors | 24 September 2020, 13:31:46 UTC | Merge pull request #4170 from apaszke:all_to_all PiperOrigin-RevId: 333502317 | 24 September 2020, 13:31:46 UTC |
acd4cc5 | Adam Paszke | 28 August 2020, 15:21:50 UTC | Allow vmapping all_to_all and implement a (slow) CPU and GPU translation This allows pmapping vmapped computations that use `all_to_all` or `pswapaxes` inside. It also includes a very slow CPU and GPU translation rule that might be useful for debugging programs locally, since XLA only implements the `AllToAll` collective on TPUs. Fixes #4141. | 24 September 2020, 12:18:59 UTC |
03970e9 | George Necula | 24 September 2020, 12:08:07 UTC | Tag tests that require omnistaging | 24 September 2020, 12:08:07 UTC |
2b61513 | George Necula | 24 September 2020, 11:24:02 UTC | [host_callback] Fixes and documentation changes for omnistaging As a nice benefit from omnistaging, the id_print are executed in program order even if the result is not ued. We keep the functions returning values for backwards compatibility, but also so the can be used inside expressions and also so that they can tap values from the backward computation. Omnistaging introduces jaxpr with no invars, which broke host_callback. | 24 September 2020, 11:24:02 UTC |
f7a09c6 | jax authors | 24 September 2020, 10:28:32 UTC | Merge pull request #4388 from apaszke:all_to_all_translation PiperOrigin-RevId: 333480692 | 24 September 2020, 10:28:32 UTC |
b8396c1 | jax authors | 24 September 2020, 10:09:12 UTC | Merge pull request #4364 from botev:api PiperOrigin-RevId: 333478420 | 24 September 2020, 10:09:12 UTC |
aa85166 | jax authors | 24 September 2020, 06:11:48 UTC | Merge pull request #4387 from gnecula:print_doc PiperOrigin-RevId: 333451138 | 24 September 2020, 06:11:48 UTC |
20c8c1a | jax authors | 24 September 2020, 05:50:49 UTC | Merge pull request #4400 from google:issue4389 PiperOrigin-RevId: 333448765 | 24 September 2020, 05:50:49 UTC |
bdcefbd | jax authors | 24 September 2020, 04:49:39 UTC | Merge pull request #4401 from google:update-pypi PiperOrigin-RevId: 333440474 | 24 September 2020, 04:49:39 UTC |
951bc75 | Matthew Johnson | 24 September 2020, 04:26:10 UTC | update version and changelog for pypi | 24 September 2020, 04:26:10 UTC |
ebf7c1b | Matthew Johnson | 24 September 2020, 04:09:44 UTC | add jax logo file | 24 September 2020, 04:09:44 UTC |
c7e0ef4 | jax authors | 24 September 2020, 03:51:27 UTC | Merge pull request #4398 from google:lift-randomness-limit PiperOrigin-RevId: 333433816 | 24 September 2020, 03:51:27 UTC |
d607164 | Matthew Johnson | 24 September 2020, 03:41:57 UTC | make_jaxpr return_shape use ShapeDtypeStruct, test | 24 September 2020, 03:41:57 UTC |
71f5f99 | Matthew Johnson | 24 September 2020, 03:15:32 UTC | skip checks in big randomness test | 24 September 2020, 03:15:32 UTC |
96f5a3c | Matthew Johnson | 24 September 2020, 02:39:22 UTC | fix test for non-omnistaging | 24 September 2020, 02:39:22 UTC |
c42d736 | Matthew Johnson | 24 September 2020, 02:37:34 UTC | remove limit on size of random arrays | 24 September 2020, 02:37:34 UTC |
0d5f15f | Adam Paszke | 23 September 2020, 10:45:23 UTC | Fix the abstract eval and translation rule for all_to_all The previous rules assumed that `split_axis == concat_axis` (i.e. that the used collective is equivalent to `pswapaxes`). Since we expose this as part of our API, we should probably make sure that we handle other cases too. Fixes #1332. | 23 September 2020, 16:22:03 UTC |
c875ab3 | jax authors | 23 September 2020, 16:11:33 UTC | Merge pull request #4391 from apaszke:axis_index_handle_list PiperOrigin-RevId: 333304709 | 23 September 2020, 16:11:33 UTC |
c9b3df3 | jax authors | 23 September 2020, 15:26:03 UTC | Merge pull request #4382 from hawkinsp:lu PiperOrigin-RevId: 333296510 | 23 September 2020, 15:26:03 UTC |
c6cd2f9 | jax authors | 23 September 2020, 15:13:34 UTC | Merge pull request #4392 from SIben:fix_eigh_flakiness_cpu PiperOrigin-RevId: 333294460 | 23 September 2020, 15:13:34 UTC |