Home » Oreps » Jax » All Releases » jax-v0.3.14 Release
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Watch
1
Commit Activity
Directory Browser
Velocity History
Releasesjax-v0.3.14
Basic Details Issues Addressed Top Contributors Directory Browser
Jax jax-v0.3.14

Jax: jax-v0.3.14 Release

Release date:
June 27, 2022
Previous version:
jax-v0.3.13 (released May 16, 2022)
Magnitude:
25,706 Diff Delta
Contributors:
52 total committers
Data confidence:
Commits:
308 impacting commits

308 Commits in this Release

Ordered by the degree to which they evolved the repo in this version.

feat: refactor code using pyupgrade
Authored May 12, 2022
[dynamic-shapes] initial support for dynamic shape typechecks
Authored June 11, 2022
Add output token for unordered effects
Authored May 17, 2022
feat: validate jit args
Authored May 6, 2022
[x64] make jax.scipy.signal compatible with strict dtype promotion.
Authored June 17, 2022
Make QDWH-eig implementation jit-table.
Authored May 18, 2022
[djax] add support for dynamic-shape outputs
Authored May 19, 2022
Add stateful for loop primitives (#10982)
Authored June 2, 2022
add support for lambda-bound dynamic shape output (iree only)
Authored May 19, 2022
[new-remat] add _scan_partial_eval_custom rule for new remat
Authored May 1, 2022
First CL to integrate `jax.Array` into `pmap`.
Authored June 17, 2022
Copybara import of the project:
Authored May 26, 2022
[x64] make linalg functions & tests compatible with strict dtype promotion
Authored June 16, 2022
Adding `jax.Array` to jax.experimental. Its pretty much the same as GDA (without the performance optimization for now).
Authored June 7, 2022
jnp.roots: better support for computation under JIT
Authored June 23, 2022
Add pjit support for `Array`. `Array` takes the same codepath as GDA so there are very little modifications to pjit. Add handlers aval, shard_args and result handlers for Array.
Authored June 10, 2022
Copybara import of the project:
Authored May 27, 2022
Add jtu.strict_promotion_if_dtypes_match utility
Authored June 16, 2022
Drop flatbuffers as a Python dependency of JAX.
Authored June 27, 2022
Allow optional out_axis_resources for jax.Array. I'll look into making this enabled for all types in a follow up CL. If out_axis_resources is _UNSPECIFIED, then we let classic spmd choose the sharding for us. If you want to use the AUTO spmd partitioner, then pass pjit.AUTO to out_axis_resources explicitly.
Authored June 11, 2022
Add an option to create a perfetto link in the JAX profiler
Authored May 25, 2022
Use JAX's distributed system for fully asynchronous checkpointing.
Authored May 18, 2022
Decompose lax.linalg.qr into two subprimitives geqrf and orgqr.
Authored May 16, 2022
Refactor `control_flow.py` into several smaller pieces
Authored June 2, 2022
[TPU] Switch the default eigendecomposition implementation on TPU to use QDWH-eig.
Authored May 27, 2022
[x64] make scipy_stats_test.py compatible with strict dtype promotion
Authored June 14, 2022
Add `jax.default_device` context manager
Authored December 21, 2021
DeviceArray: Improve support for copy, deepcopy, and pickle
Authored May 17, 2022
Add collect_profile script
Authored June 3, 2022
Check sharding in pmap for `jax.Array`.
Authored June 22, 2022
feat: support subset of scatters with no XLA
Authored May 14, 2022
Using etils(gfile) to support gcs buckets and file system for persistent compilation caching
Authored May 19, 2022
Add jax_numpy_dtype_promotion='strict' mode
Authored May 26, 2022
Enable python callback on GPU
Authored June 1, 2022
Add a public facing `named_scope` function to allow adding to the name stack.
Authored May 25, 2022
[x64] make fft functionality compatible with strict dtype promotion
Authored June 15, 2022
Unbreak jax.distributed initialization.
Authored June 7, 2022
Add Array support to xmap. Just using the GDA path.
Authored June 13, 2022
[x64] deprecate unsafe type casting in scatter-update operations
Authored June 9, 2022
Add a batched QR decomposition implementation on GPU.
Authored May 18, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.18-rc Released September 26, 2022
18,442 Ξ”
jax-v0.3.17 Released August 31, 2022
20,878 Ξ”
jax-v0.3.16 Released August 11, 2022
16,659 Ξ”
jax-v0.3.15 Released July 22, 2022
0 Ξ”
jax-v0.3.15-rc Released July 22, 2022
18,821 Ξ”
jax-v0.3.14 Released June 27, 2022
25,706 Ξ”
jax-v0.3.13 Released May 16, 2022
190 Ξ”
jax-v0.3.12 Released May 16, 2022
13 Ξ”
jax-v0.3.11 Released May 14, 2022
9,164 Ξ”
jax-v0.3.10 Released May 4, 2022
1,645 Ξ”

Top Contributors in jax-v0.3.14

jakevdp
yashk2810
mattjj
hawkinsp
sharadmv
JeppeKlitgaard
LenaMartens
tlu7
gnecula
skye

Directory Browser for jax-v0.3.14

We haven't yet finished calculating and confirming the files and directories changed in this release. Please check back soon.

GitClear uses cookies to ensure you get the best experience on our website. Learn more