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

Jax: jaxlib-v0.1.75 Release

Release date:
December 7, 2021
Previous version:
jaxlib-v0.1.74 (released November 17, 2021)
Magnitude:
17,697 Diff Delta
Contributors:
20 total committers
Data confidence:
Commits:
171 impacting commits

171 Commits in this Release

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

Split jax.interpreters.xla up into three pieces:
Authored November 22, 2021
Merge most of the MLIR JIT dispatch logic into the common primitive and JIT computation path.
Authored November 29, 2021
Use imports relative to the `jax` package consistently, rather than `.`-relative imports.
Authored November 24, 2021
[shape_poly] Refactor shape_poly to be independent of TF.
Authored November 17, 2021
add skeleton checkify transformation
Authored October 29, 2021
[MLIR] Add more parallelism and sharding support via MLIR.
Authored December 3, 2021
Split slice/update_slice/gather/scatter out of jax._src.lax.lax into jax._src.lax.slicing.
Authored November 24, 2021
Copybara import of the project:
Authored December 1, 2021
Increase minimum jaxlib version to 0.1.74.
Authored November 18, 2021
[sparse] refactor jax.experimental.sparse
Authored December 2, 2021
refactor pmap staging, lowering, and compilation
Authored November 19, 2021
[sparse] spdot_general: implement many more cases
Authored November 18, 2021
Make lax.reduce_window variadic.
Authored November 22, 2021
Lift contiguous mesh requirement for GDAs. If all inputs are GDAs, the mesh can be non-contiguous.
Authored December 3, 2021
[MLIR] Fix CPU test failures for MLIR lowering.
Authored November 30, 2021
Consolidate more XLA-lowering logic between jit, pmap, and xmap.
Authored November 30, 2021
[sparse] Improve type safety of cusparse lowerings
Authored November 17, 2021
Adding support for a special value for in_axis_resources (`pjit.FROM_GSDA`) when GSDA is an input.
Authored November 19, 2021
Fix a number of bugs in MLIR translation rules.
Authored November 17, 2021
[jax2tf] Force TF compilation for code under jax.jit.
Authored September 7, 2021
Split windowed reductions and their gradients into a separate file inside the lax implementation.
Authored November 18, 2021
generalize jaxpr simplification machinery
Authored November 16, 2021
Move serialization/de-serialization of GDA into jax.
Authored December 7, 2021
Internal change
Authored December 6, 2021
Move dtype canonicalization out of core.AbstractValue subclasses.
Authored December 7, 2021
[MLIR] Make jit(pmap(...)) work in the MLIR lowering.
Authored November 30, 2021
Improve logics for numerical comparison
Authored November 18, 2021
[x64] make jax.numpy functionality respect default dtypes
Authored December 1, 2021
[shape_poly] Refactor arange and image_resize for shape polymorphism
Authored November 5, 2021
[x64] Make autodiff respect weak types
Authored November 23, 2021
factor pmap compilation into lowering and compilation separately
Authored November 3, 2021
Split `aval_to_result_handler` into `local_aval_to_result_handler` and `global_aval_to_result_handler` because `aval_to_result_handler` was shared by pmap, pjit and xmap. This led to parameters having optional values (None) and hacks were introduced to get around that.
Authored December 6, 2021
Split convolution functions out of jax._src.lax.lax and into a separate module (jax._src.lax.convolution).
Authored November 23, 2021
Refactor the lowering of remat.
Authored November 29, 2021
Express xmap tile/untile logic in lax via xla.lower_fun().
Authored December 2, 2021
[sparse] specify operand layouts in cusparse.py
Authored November 19, 2021
[jax2tf] Improved error checking for call_tf for functions with dynamic
Authored December 1, 2021
[sparse] add BCOO.sum_duplicates() with nse option
Authored November 16, 2021
[x64] regularize dtype helpers
Authored November 22, 2021
Refactor uses of xla.call_translations to use xla.register_translation.
Authored December 1, 2021

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.3.7 Released April 15, 2022
10,780 Ξ”
jaxlib-v0.3.5 Released April 7, 2022
9,400 Ξ”
jaxlib-v0.3.2 Released March 15, 2022
19,405 Ξ”
jaxlib-v0.3.0 Released February 10, 2022
8,876 Ξ”
jaxlib-v0.1.76 Released January 27, 2022
15,529 Ξ”
jaxlib-v0.1.75 Released December 7, 2021
17,697 Ξ”
jaxlib-v0.1.74 Released November 17, 2021
14,226 Ξ”
jaxlib-v0.1.73 Released October 18, 2021
0 Ξ”
jaxlib-v0.1.72 Released October 12, 2021
11,692 Ξ”
jaxlib-v0.1.71 Released September 1, 2021
0 Ξ”

Top Contributors in jaxlib-v0.1.75

hawkinsp
jakevdp
yashk2810
gnecula
mattjj
froystig
a-googler
marcvanzee
rsepassi
tlu7

Directory Browser for jaxlib-v0.1.75

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

Release Notes Published

  • New features:
    • Support for python 3.10.
GitClear uses cookies to ensure you get the best experience on our website. Learn more