Home » Oreps » Jax » All Releases » jaxlib-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
Releasesjaxlib-v0.3.14
Basic Details Issues Addressed Top Contributors Directory Browser
Jax jaxlib-v0.3.14

Jax: jaxlib-v0.3.14 Release

Release date:
June 27, 2022
Previous version:
jaxlib-v0.3.10 (released May 4, 2022)
Magnitude:
33,978 Diff Delta
Contributors:
62 total committers
Data confidence:
Commits:
398 impacting commits

398 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
Merge remaining CUDA and ROCM Python code.
Authored May 6, 2022
[dynamic-shapes] initial support for dynamic shape typechecks
Authored June 11, 2022
Add sequencing effects design note
Authored May 10, 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
Enable print in while loops/scan
Authored May 4, 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
add scan dce rule tests, fix bugs
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
improve partial_eval_jaxpr_custom
Authored May 7, 2022
[JAX] Split the "gpu" platform in internal JAX usage into separate "cuda" and "rocm" platforms.
Authored May 5, 2022
[sparse] add bcoo_update_layout utility
Authored May 5, 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
add core.closed_call_p
Authored May 14, 2022
Copybara import of the project:
Authored May 27, 2022
Add jtu.strict_promotion_if_dtypes_match utility
Authored June 16, 2022
Rollback of PR #10393 "Improve performance of array integer indexing"
Authored May 10, 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
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
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

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.3.24 Released November 4, 2022
23,851 Ξ”
jaxlib-v0.3.22 Released October 11, 2022
12,900 Ξ”
jaxlib-v0.3.20 Released September 28, 2022
55,337 Ξ”
jaxlib-v0.3.15 Released July 22, 2022
16,976 Ξ”
jaxlib-v0.3.14 Released June 27, 2022
33,978 Ξ”
jaxlib-v0.3.10 Released May 4, 2022
13,335 Ξ”
jaxlib-v0.3.7 Released April 15, 2022
10,780 Ξ”
jaxlib-v0.3.5 Released April 7, 2022
9,385 Ξ”
jaxlib-v0.3.2 Released March 15, 2022
19,405 Ξ”
jaxlib-v0.3.0 Released February 10, 2022
6,231 Ξ”

Top Contributors in jaxlib-v0.3.14

jakevdp
hawkinsp
sharadmv
mattjj
yashk2810
JeppeKlitgaard
tlu7
LenaMartens
gnecula
lgeiger

Directory Browser for jaxlib-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