| 7ad5126 | Skye Wanderman-Milne | 17 March 2021, 01:26:27 UTC | Update WORKSPACE for jaxlib 0.1.63 release, take 2 PiperOrigin-RevId: 363315929 | 17 March 2021, 01:26:52 UTC |
| 0a84db5 | jax authors | 16 March 2021, 22:20:40 UTC | Merge pull request #6068 from jakevdp:fix-result-type PiperOrigin-RevId: 363282198 | 16 March 2021, 22:20:40 UTC |
| 6e1cd39 | jax authors | 16 March 2021, 21:45:24 UTC | Merge pull request #6075 from jakevdp:fix-formatting PiperOrigin-RevId: 363274347 | 16 March 2021, 21:45:24 UTC |
| 4d8e954 | Jake VanderPlas | 15 March 2021, 23:50:07 UTC | Sharp Bits: fix formatting of code blocks within list | 16 March 2021, 21:21:21 UTC |
| 3b7de31 | jax authors | 16 March 2021, 20:58:15 UTC | Merge pull request #6087 from jakevdp:pypi-extras PiperOrigin-RevId: 363263098 | 16 March 2021, 20:58:15 UTC |
| d6408a4 | Jake VanderPlas | 16 March 2021, 18:40:46 UTC | Add extras_require to setup.py | 16 March 2021, 20:23:46 UTC |
| d2d7ecf | jax authors | 16 March 2021, 20:21:52 UTC | Merge pull request #6082 from google:minjaxlib PiperOrigin-RevId: 363255561 | 16 March 2021, 20:21:52 UTC |
| 328930b | Peter Hawkins | 16 March 2021, 16:13:41 UTC | Increase minimum jaxlib version to 0.1.62. | 16 March 2021, 19:11:36 UTC |
| d326b07 | jax authors | 16 March 2021, 19:10:03 UTC | Merge pull request #6086 from hawkinsp:numpy PiperOrigin-RevId: 363239393 | 16 March 2021, 19:10:03 UTC |
| 2bf7dbc | jax authors | 16 March 2021, 18:28:46 UTC | Merge pull request #5736 from tberghammer:changelist/357690519 PiperOrigin-RevId: 363229118 | 16 March 2021, 18:28:46 UTC |
| 48a7b15 | Peter Hawkins | 16 March 2021, 17:46:13 UTC | Fix test failure with NumPy 1.20. Fixes #6083 | 16 March 2021, 17:46:13 UTC |
| 265a663 | jax authors | 16 March 2021, 16:51:55 UTC | Merge pull request #6084 from skye:workspace PiperOrigin-RevId: 363205601 | 16 March 2021, 16:51:55 UTC |
| b319d23 | Skye Wanderman-Milne | 16 March 2021, 16:43:22 UTC | Update WORKSPACE for jaxlib 0.1.63 release | 16 March 2021, 16:43:22 UTC |
| 2ea5261 | Tamas Berghammer | 16 February 2021, 12:31:01 UTC | Add new lax.rng_bit_generator primitive The new primitive provides access to the RngBitGenerator HLO (https://www.tensorflow.org/xla/operation_semantics#rngbitgenerator) | 16 March 2021, 16:30:09 UTC |
| 2d148a3 | jax authors | 16 March 2021, 15:03:23 UTC | Merge pull request #6078 from jacobaustin123:master PiperOrigin-RevId: 363183679 | 16 March 2021, 15:03:23 UTC |
| 25704a0 | jax authors | 16 March 2021, 11:14:48 UTC | Merge pull request #6081 from gnecula:jax2tf_examples PiperOrigin-RevId: 363151280 | 16 March 2021, 11:14:48 UTC |
| f67aeea | George Necula | 16 March 2021, 11:01:16 UTC | Fix the output directory | 16 March 2021, 11:01:16 UTC |
| 53d21b2 | jax authors | 16 March 2021, 10:42:02 UTC | Merge pull request #6077 from hawkinsp:jaxlibimport PiperOrigin-RevId: 363147337 | 16 March 2021, 10:42:02 UTC |
| 840d516 | George Necula | 16 March 2021, 10:38:57 UTC | [jax2tf] Removed more traces of support for batch polymorphism See issue #6080 * Also cleanup the examples | 16 March 2021, 10:38:57 UTC |
| 9d28b67 | Jacob Austin | 16 March 2021, 03:26:31 UTC | Fixed two small typos in jax.lax. | 16 March 2021, 03:26:31 UTC |
| 3da12db | jax authors | 16 March 2021, 01:24:44 UTC | Merge pull request #6048 from skye:debug_nans PiperOrigin-RevId: 363078153 | 16 March 2021, 01:24:44 UTC |
| ee53eeb | Peter Hawkins | 16 March 2021, 01:07:59 UTC | Add helpful message when `import jaxlib` fails. | 16 March 2021, 01:07:59 UTC |
| b3165bb | jax authors | 16 March 2021, 00:22:27 UTC | Merge pull request #6064 from ebuehrle:patch-1 PiperOrigin-RevId: 363067775 | 16 March 2021, 00:22:27 UTC |
| b0c5fba | Jake VanderPlas | 15 March 2021, 21:38:14 UTC | BUG: fix jnp.result_type for non-canonical weak types | 15 March 2021, 21:38:14 UTC |
| 63c06ef | Peter Hawkins | 15 March 2021, 19:29:44 UTC | [JAX] Add a .weak_type attribute to C++ array objects. Use .weak_type instead of parsing avals from C++. Inspecting Python objects unnecessarily is slow. In addition we were building a Python bool object that we didn't need to build (`py::cast<py::bool_>` instead of `py::cast<bool>`). Benchmarks on my workstation: ``` name old time/op new time/op delta jit_trivial_dispatch 44.9µs ± 1% 44.3µs ± 0% -1.37% (p=0.008 n=5+5) jit_trivial 46.2µs ± 0% 45.6µs ± 0% -1.39% (p=0.008 n=5+5) jit_simple_dispatch 17.7µs ± 2% 16.6µs ± 1% -6.37% (p=0.008 n=5+5) jit_simple 18.5µs ± 5% 17.3µs ± 1% -6.54% (p=0.008 n=5+5) jit_simple_many_args_dispatch_10 26.6µs ± 1% 22.6µs ± 2% -15.12% (p=0.008 n=5+5) jit_simple_many_args_10 27.9µs ± 3% 24.6µs ± 4% -12.00% (p=0.008 n=5+5) jit_simple_many_args_dispatch_100 107µs ± 1% 75µs ± 1% -29.85% (p=0.008 n=5+5) jit_simple_many_args_100 108µs ± 1% 76µs ± 0% -29.66% (p=0.008 n=5+5) jit_simple_many_args_dispatch_1000 1.01ms ± 1% 0.69ms ± 2% -31.72% (p=0.008 n=5+5) jit_simple_many_args_1000 1.03ms ± 1% 0.71ms ± 2% -30.77% (p=0.008 n=5+5) jit_simple_many_args_dispatch_2000 2.09ms ± 1% 1.43ms ± 3% -31.78% (p=0.008 n=5+5) jit_simple_many_args_2000 2.08ms ± 1% 1.44ms ± 4% -30.77% (p=0.008 n=5+5) jit_dispatch_without_transfer 1.41ms ± 1% 1.43ms ± 6% ~ (p=1.000 n=5+5) jit_dispatch_with_transfer 1.40ms ± 1% 1.40ms ± 1% ~ (p=1.000 n=5+5) ``` PiperOrigin-RevId: 363002879 | 15 March 2021, 19:30:15 UTC |
| 9a2a1ad | Peter Hawkins | 15 March 2021, 17:19:31 UTC | [JAX] Enable C++ device arrays by default. [XLA:Python] Relax constraints on .aval and ._device attributes on C++ buffer objects. The constraints cause more problems than they solve. Switch _device to be a C++ attribute rather than a Python attribute. This avoids some unnecessary Python attribute parsing in the JIT dispatch path. Change PyBuffer objects to call themselves `DeviceArray` in Python so as not to surprise JAX users. PiperOrigin-RevId: 362969997 | 15 March 2021, 17:20:22 UTC |
| 2e90e72 | ebuehrle | 15 March 2021, 10:24:32 UTC | Fix code typo | 15 March 2021, 16:20:47 UTC |
| 80966fe | jax authors | 15 March 2021, 16:19:52 UTC | Merge pull request #6018 from jakevdp:conv-elem-type PiperOrigin-RevId: 362956294 | 15 March 2021, 16:19:52 UTC |
| 3fb6a11 | jax authors | 15 March 2021, 13:56:14 UTC | Merge pull request #6057 from inailuig:fix-complex-normal PiperOrigin-RevId: 362930693 | 15 March 2021, 13:56:14 UTC |
| 1ad99d3 | jax authors | 15 March 2021, 10:10:41 UTC | Merge pull request #5999 from sharadmv:callback-scan PiperOrigin-RevId: 362898757 | 15 March 2021, 10:10:41 UTC |
| d78fe6b | Clemens Giuliani | 14 March 2021, 21:23:05 UTC | fix the dtype of complex jax.random.normal and add a regression test for it | 14 March 2021, 21:28:38 UTC |
| c56649a | Skye Wanderman-Milne | 12 March 2021, 19:27:42 UTC | Make jax_debug_nans and jax_debug_infs work with pmap, xmap, and pjit. Note that unlike in the jit case, this doesn't rerun the function in op-by-op mode when it finds a nan, since we don't have op-by-op parallel execution yet :) This change doesn't appear to regress performance: ``` ---------Benchmark summary for pmap_shard_outputs--------- nouts nshards mean %std relative mean/baseline ------- --------- --------- -------- ---------- --------------- 10 8 0.105598 5.06671 1 1.00693 100 8 0.287756 0.870751 2.72502 0.973204 500 8 1.20119 0.823624 11.3752 0.955185 1000 8 2.56071 0 24.2497 0.983063 5000 8 12.909 0 122.247 0.965925 100 2 0.173727 5.15115 1.64518 0.98918 100 4 0.207774 3.71411 1.9676 0.955849 100 8 0.286103 1.60243 2.70937 0.971869 100 100 2.34168 0 22.1755 0.904475 100 500 15.9558 0 151.1 1.00483 ``` Fixes #6044 | 13 March 2021, 00:22:55 UTC |
| 04bf02a | Jake VanderPlas | 12 March 2021, 23:26:06 UTC | convert_element_type: don't canonicalize old_dtype | 12 March 2021, 23:26:06 UTC |
| 8d3b4ac | jax authors | 12 March 2021, 21:38:13 UTC | Merge pull request #6028 from jakevdp:transpose PiperOrigin-RevId: 362590852 | 12 March 2021, 21:38:13 UTC |
| 77c1f31 | jax authors | 12 March 2021, 19:35:44 UTC | Merge pull request #5966 from mtsokol:jax-numpy-where-keyword PiperOrigin-RevId: 362565473 | 12 March 2021, 19:35:44 UTC |
| a42e653 | jax authors | 12 March 2021, 19:19:49 UTC | Merge pull request #6029 from jakevdp:timeout-minutes PiperOrigin-RevId: 362561981 | 12 March 2021, 19:19:49 UTC |
| ed4c944 | Jake VanderPlas | 11 March 2021, 17:30:03 UTC | jnp.array.transpose: support positional axis arguments | 12 March 2021, 19:16:50 UTC |
| 60dcd0d | Jake VanderPlas | 11 March 2021, 19:55:50 UTC | Set reasonable timeouts for github actions jobs | 12 March 2021, 17:31:43 UTC |
| ea07d41 | jax authors | 12 March 2021, 17:29:21 UTC | Merge pull request #6041 from tomhennigan:changelist/362481121 PiperOrigin-RevId: 362537513 | 12 March 2021, 17:29:21 UTC |
| d743aa5 | Mateusz Sokół | 06 March 2021, 10:53:01 UTC | Added 'where' keyword to 'jnp.{mean, var, std}' | 12 March 2021, 16:57:17 UTC |
| 4f74b33 | Tom Hennigan | 12 March 2021, 16:38:29 UTC | Update README.md Co-authored-by: 8bitmp3 <19637339+8bitmp3@users.noreply.github.com> | 12 March 2021, 16:38:29 UTC |
| 9785230 | jax authors | 12 March 2021, 14:02:50 UTC | Merge pull request #6043 from hawkinsp:pocketfft PiperOrigin-RevId: 362504596 | 12 March 2021, 14:02:50 UTC |
| 1ed321c | Peter Hawkins | 12 March 2021, 13:46:01 UTC | Update PocketFFT version to fix crash due to undersized aligned allocations. | 12 March 2021, 13:46:01 UTC |
| 0fd83a3 | Tom Hennigan | 12 March 2021, 10:40:17 UTC | Link to DeepMind JAX blog post. | 12 March 2021, 12:01:20 UTC |
| c7ebc3e | jax authors | 11 March 2021, 23:22:38 UTC | Merge pull request #6032 from skye:grpc_env_var PiperOrigin-RevId: 362389283 | 11 March 2021, 23:22:38 UTC |
| ee8ecb0 | jax authors | 11 March 2021, 23:15:53 UTC | Merge pull request #6015 from sethvargo:patch-1 PiperOrigin-RevId: 362387994 | 11 March 2021, 23:15:53 UTC |
| 5cb5056 | Skye Wanderman-Milne | 11 March 2021, 22:52:54 UTC | Suppress gRPC log spam on Cloud TPU. | 11 March 2021, 22:52:54 UTC |
| 077793c | jax authors | 11 March 2021, 21:23:53 UTC | Merge pull request #6019 from skye:examples_test PiperOrigin-RevId: 362363477 | 11 March 2021, 21:23:53 UTC |
| 65ee604 | jax authors | 11 March 2021, 21:11:21 UTC | Merge pull request #6020 from jamestwebber:patch-1 PiperOrigin-RevId: 362360958 | 11 March 2021, 21:11:21 UTC |
| cf4f445 | James Webber | 11 March 2021, 19:24:41 UTC | sync markdown | 11 March 2021, 19:24:41 UTC |
| 6b923a9 | jax authors | 11 March 2021, 18:48:43 UTC | Merge pull request #6026 from Chillee:patch-1 PiperOrigin-RevId: 362328216 | 11 March 2021, 18:48:43 UTC |
| abdee43 | Horace He | 11 March 2021, 10:56:52 UTC | Update 04-advanced-autodiff.md | 11 March 2021, 10:56:52 UTC |
| e38d272 | Horace He | 11 March 2021, 10:55:48 UTC | Update 04-advanced-autodiff.ipynb | 11 March 2021, 10:55:48 UTC |
| 7ef86ef | jax authors | 11 March 2021, 07:19:44 UTC | Merge pull request #6013 from zhangqiaorjc:new_dlpack_fix PiperOrigin-RevId: 362224066 | 11 March 2021, 07:19:44 UTC |
| cd61997 | Skye Wanderman-Milne | 11 March 2021, 00:34:37 UTC | Adjust precision in examples/kernel_lsq.py and corresponding test. This is important on TPU. | 11 March 2021, 03:03:43 UTC |
| 3a99e58 | James Webber | 11 March 2021, 01:09:54 UTC | (hopefully) fix broken link | 11 March 2021, 01:09:54 UTC |
| 72a3036 | James Bradbury | 11 March 2021, 00:08:58 UTC | Hotfix for another assertion that's too strict about named shapes PiperOrigin-RevId: 362164157 | 11 March 2021, 00:09:37 UTC |
| 9b7aa37 | Seth Vargo | 10 March 2021, 22:55:21 UTC | Update ci-build.yaml | 10 March 2021, 22:55:21 UTC |
| cf9b77f | jax authors | 10 March 2021, 22:36:55 UTC | Merge pull request #5998 from zhangqiaorjc:dev_put_count PiperOrigin-RevId: 362143966 | 10 March 2021, 22:36:55 UTC |
| e523ae0 | Qiao Zhang | 08 September 2020, 21:28:26 UTC | Add a test that fails due to unsupported strides. - test fails during dlpack tensor to buffer conversion - XLA layout does not support arbitrary dlpack strides - users should explicit materialze such tensors by making a copy | 10 March 2021, 22:19:21 UTC |
| ecbc134 | Matthew Johnson | 10 March 2021, 22:16:48 UTC | fix typo | 10 March 2021, 22:16:48 UTC |
| 1048ebb | Seth Vargo | 10 March 2021, 22:14:48 UTC | Update ci-build.yaml | 10 March 2021, 22:14:48 UTC |
| e550dda | Seth Vargo | 10 March 2021, 21:26:01 UTC | Improve Actions hygiene 👋 hello there! I'm a fellow Googler who works on projects that leverage GitHub Actions for CI/CD. Recently I noticed a large increase in our queue time, and I've tracked it down to the [limit of 180 concurrent jobs](https://docs.github.com/en/actions/reference/usage-limits-billing-and-administration) for an organization. To help be better citizens, I'm proposing changes across a few repositories that will reduce GitHub Actions hours and consumption. I hope these changes are reasonable and I'm happy to talk through them in more detail. - **(you were already doing this, thank you!**) Only run GitHub Actions for pushes and PRs against the main branch of the repository. If your team uses a forking model, this change will not affect you. If your team pushes branches to the repository directly, this changes actions to only run against the primary branches or if you open a Pull Request against a primary branch. - For long-running jobs (especially tests), I added the "Cancel previous" workflow. This is very helpful to prevent a large queue backlog when you are doing rapid development and pushing multiple commits. Without this, GitHub Actions' default behavior is to run all actions on all commits. There are other changes you could make, depending on your project (but I'm not an expert): - If you have tests that should only run when a subset of code changes, consider gating your workflow to particular file paths. For example, we have some jobs that do Terraform linting, but [they only run when Terraform files are changed](https://github.com/google/exposure-notifications-verification-server/blob/c4f59fee71042cf668747e599e7c769fca736554/.github/workflows/terraform.yml#L3-L11). Hopefully these changes are not too controversial and also hopefully you can see how this would reduce actions consumption to be good citizens to fellow Googlers. If you have any questions, feel free to respond here or ping me on chat. Thank you! | 10 March 2021, 21:26:01 UTC |
| 61041cb | jax authors | 10 March 2021, 20:49:55 UTC | Merge pull request #6010 from hawkinsp:issue4690 PiperOrigin-RevId: 362120286 | 10 March 2021, 20:49:55 UTC |
| 62a726d | Peter Hawkins | 10 March 2021, 14:57:31 UTC | Add workaround for SelectAndScatter padding bug on CPU and GPU. | 10 March 2021, 20:25:32 UTC |
| d11bba9 | Matthew Johnson | 10 March 2021, 20:20:36 UTC | tweak broken tests | 10 March 2021, 20:20:36 UTC |
| c9c89c4 | jax authors | 10 March 2021, 18:32:56 UTC | Merge pull request #5997 from jakevdp:fix-piecewise PiperOrigin-RevId: 362085337 | 10 March 2021, 18:32:56 UTC |
| b675905 | jax authors | 10 March 2021, 18:14:08 UTC | Merge pull request #6012 from hawkinsp:lazy3 PiperOrigin-RevId: 362079742 | 10 March 2021, 18:14:08 UTC |
| d0de749 | jax authors | 10 March 2021, 18:06:21 UTC | Merge pull request #5996 from skye:cudnn_versions PiperOrigin-RevId: 362077738 | 10 March 2021, 18:06:21 UTC |
| 5098caf | jax authors | 10 March 2021, 17:49:40 UTC | Merge pull request #5984 from skye:cloud_tpu_env_vars PiperOrigin-RevId: 362072957 | 10 March 2021, 17:49:40 UTC |
| c32d1e5 | Skye Wanderman-Milne | 05 March 2021, 22:57:36 UTC | Automatically initialize Cloud TPU topology env vars if running on a Cloud TPU VM. This removes the need to manually set these env vars when running on a Cloud TPU pod slice. | 10 March 2021, 17:15:31 UTC |
| 0acafcf | jax authors | 10 March 2021, 16:59:12 UTC | Merge pull request #6011 from hawkinsp:token PiperOrigin-RevId: 362060386 | 10 March 2021, 16:59:12 UTC |
| 5d6ff8b | Peter Hawkins | 10 March 2021, 16:34:42 UTC | Readd omnistaging friendly versions of large constant tests. | 10 March 2021, 16:34:42 UTC |
| 0034939 | Peter Hawkins | 10 March 2021, 15:18:38 UTC | Fix crash returning a Token from a jit computation on GPU. Calling .numpy_dtype() doesn't work on tokens. But we don't need a numpy dtype here, an XLA dtype works just as well. | 10 March 2021, 15:18:38 UTC |
| 3e45a83 | jax authors | 10 March 2021, 14:27:20 UTC | Merge pull request #6002 from hawkinsp:lazy3 PiperOrigin-RevId: 362034533 | 10 March 2021, 14:27:20 UTC |
| ddaef19 | Sharad Vikram | 09 March 2021, 23:21:07 UTC | Add scan and while rule for jax.experimental.callback transformation | 10 March 2021, 03:46:16 UTC |
| 23b82b9 | jax authors | 10 March 2021, 02:56:54 UTC | Merge pull request #6001 from skye:jaxlib_version PiperOrigin-RevId: 361948443 | 10 March 2021, 02:56:54 UTC |
| 140c0ac | Peter Hawkins | 10 March 2021, 02:31:17 UTC | Remove the JAX lazy sublanguage. Back in the mists of time, before omnistaging landed in JAX, we used lazy expressions to avoid materializing large constants inside `jit` computations. Omnistaging, which means that computations that are in the dynamic scope of a `jit` are staged into the `jit` computation, has subsumed most of the reasons for laziness to exist, and this PR removes the laziness support for simplicity. At the time of this PR, laziness is used only for broadcasts and transposes in eager mode (i.e., outside a `jit`). This allows us to: a) fuse together multiple broadcasts and transposes, and b) if a lazy expression is lexically captured by a `jit` computation, we can avoid materializing it in its expanded form. It is not clear that laziness has sufficient power to weight ratio to continue to exist, and it is making other work on improving JAX dispatch times more difficult. As a result, this PR removes laziness to unblock that work; if we want laziness again we would want to reimplement it in C++ anyway. | 10 March 2021, 02:40:46 UTC |
| f06bb9a | Skye Wanderman-Milne | 10 March 2021, 01:55:40 UTC | Update jaxlib version etc. | 10 March 2021, 01:55:40 UTC |
| 23099f6 | jax authors | 10 March 2021, 00:27:37 UTC | Merge pull request #5835 from google:awn-abstract-eval PiperOrigin-RevId: 361923732 | 10 March 2021, 00:27:37 UTC |
| 9577860 | Qiao Zhang | 09 March 2021, 22:22:27 UTC | Add jtu.count_device_put for tests to count device_put. | 09 March 2021, 22:45:01 UTC |
| 2b9ffb1 | Matthew Johnson | 09 March 2021, 03:49:10 UTC | make axis_index bind respect dynamic traces | 09 March 2021, 21:51:12 UTC |
| 47f17f0 | James Bradbury | 04 March 2021, 12:13:04 UTC | add (broken) test | 09 March 2021, 21:51:12 UTC |
| a8b8246 | James Bradbury | 09 March 2021, 21:51:09 UTC | add some todos | 09 March 2021, 21:51:09 UTC |
| 1f17fa6 | Roy Frostig | 11 February 2021, 01:46:00 UTC | avoid modifying global pytype mappings in named-shape jaxpr test | 09 March 2021, 21:48:26 UTC |
| e779ed8 | Roy Frostig | 10 February 2021, 19:04:19 UTC | simplify standard named_shape_rule Co-authored-by: Matthew Johnson <mattjj@google.com> | 09 March 2021, 21:48:26 UTC |
| c622422 | James Bradbury | 09 March 2021, 21:48:15 UTC | [avals with names] Propagate presence of name (mapped) vs absence (replicated) in abstract eval based on existing batching rules | 09 March 2021, 21:48:15 UTC |
| ba0f785 | James Bradbury | 24 February 2021, 19:03:11 UTC | allow named axes on while_loop condition aval | 09 March 2021, 21:45:17 UTC |
| dbdb189 | Jake VanderPlas | 09 March 2021, 21:25:38 UTC | jnp.piecewise: support scalar inputs | 09 March 2021, 21:25:38 UTC |
| 6515b5f | jax authors | 09 March 2021, 19:22:18 UTC | Merge pull request #5977 from apaszke:xmap-with-control-flow PiperOrigin-RevId: 361854852 | 09 March 2021, 19:22:18 UTC |
| f1ba3bc | jax authors | 09 March 2021, 19:12:39 UTC | Merge pull request #5990 from jakevdp:fix-power PiperOrigin-RevId: 361852538 | 09 March 2021, 19:12:39 UTC |
| 55c75c8 | Skye Wanderman-Milne | 09 March 2021, 19:07:00 UTC | Update cuDNN versions to highest available. The TF build depends on newer versions than we previously used. | 09 March 2021, 19:07:00 UTC |
| b0d14fd | jax authors | 09 March 2021, 18:29:12 UTC | Merge pull request #5951 from apaszke:revive-all-to-all PiperOrigin-RevId: 361841703 | 09 March 2021, 18:29:12 UTC |
| 0c86c1f | Jake VanderPlas | 09 March 2021, 17:36:41 UTC | jnp.power: fix overflow case for x1=0 | 09 March 2021, 17:36:41 UTC |
| 0b88b0e | jax authors | 09 March 2021, 04:50:15 UTC | Merge pull request #5856 from google:autodidax PiperOrigin-RevId: 361722343 | 09 March 2021, 04:50:15 UTC |
| 8c7b8df | jax authors | 09 March 2021, 02:44:34 UTC | Merge pull request #5978 from jakevdp:poisson-zero PiperOrigin-RevId: 361707278 | 09 March 2021, 02:44:34 UTC |
| a61c43c | jax authors | 09 March 2021, 01:13:52 UTC | Merge pull request #5983 from skye:enable_tpu PiperOrigin-RevId: 361692916 | 09 March 2021, 01:13:52 UTC |
| 591a484 | jax authors | 09 March 2021, 00:34:39 UTC | Merge pull request #5908 from jakevdp:transforms-doc PiperOrigin-RevId: 361685827 | 09 March 2021, 00:34:39 UTC |
| 749ad95 | Jake VanderPlas | 09 March 2021, 00:25:04 UTC | DOC: add transformations doc to HTML & reorganize contents | 09 March 2021, 00:25:04 UTC |
| 7b7a2a1 | Skye Wanderman-Milne | 09 March 2021, 00:19:19 UTC | Don't build with Cloud TPU support for GPU or Mac wheels. This should make our builds simpler and less failure-prone. | 09 March 2021, 00:19:19 UTC |
