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

Jax: jaxlib-v0.1.76 Release

Release date:
January 27, 2022
Previous version:
jaxlib-v0.1.75 (released December 7, 2021)
Magnitude:
15,529 Diff Delta
Contributors:
30 total committers
Data confidence:
Commits:
179 impacting commits

179 Commits in this Release

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

[MLIR] Change signature of lowering rules.
Authored December 16, 2021
[jax2tf] Simplify the shape_poly support.
Authored December 16, 2021
add staging logic for polymorphic shapes in jaxprs
Authored December 11, 2021
introduce custom_transpose and a corresponding primitive
Authored January 7, 2022
Implement SciPy's RegularGridInterpolator
Authored November 19, 2021
Speed up check_jaxpr().
Authored January 18, 2022
Checkify: add way to disable categories of errors.
Authored January 10, 2022
fix a custom_vjp post_process bug, related cleanups
Authored December 11, 2021
[sparse] generalize metadata argument in BCOO primitives
Authored January 7, 2022
Add build time support for AOT compilation to TF graphs.
Authored December 20, 2021
PartitionSpec: allow partially specified sharding
Authored January 13, 2022
generate bit widths other than 32 in `lax.rng_bit_generator`
Authored January 21, 2022
Add a `fast_path` to GDA for faster GDA construction. Local indices
Authored December 15, 2021
Optimizations for GDA to make creating GDA faster.
Authored January 13, 2022
[MLIR] Add a `@mlir.cache_lowering` decorator that lowers a primitive out-of-line as a reusable function.
Authored December 16, 2021
linalg_test: disable implicit rank promotion
Authored January 26, 2022
checkify: fix and test post_process_call/map
Authored January 19, 2022
Benchmarks for GDA. Also move `create_global_mesh` to test_utils since it was replicated in a lot of places.
Authored January 11, 2022
enable custom transformation "stacking"
Authored January 11, 2022
Add NaN checkify rule to all lax primitives.
Authored December 16, 2021
introduce custom_vmap and a corresponding custom_vmap_p primitive
Authored November 12, 2021
custom batching vmap tests
Authored December 31, 2021
[JAX] Add support for generating "equation profiles" in JAX.
Authored January 12, 2022
Canonicalize parsed partition spec before passing to lower_mesh_computation. Creates a new data structure `CanonicalizedParsedPartitionSpec` which strips empty tuples from the end of parsed partitions to canonicalize the specs so that `P(None)` and `None` for example in in_axis_resources are equivalent.
Authored January 14, 2022
Make lapack symbols strong in lapack_kernels
Authored January 26, 2022
Split lax_control_flow_test into three separate tests.
Authored January 14, 2022
Internal change
Authored January 8, 2022
Re-applying #9136 after it was rolled back.
Authored January 9, 2022
lax.sort: regularize handling of -0.0 and -NaN
Authored January 13, 2022
simpler jaxpr eqn params to bind params conversion
Authored January 8, 2022
jnp.array: use jax-style promotion for list inputs
Authored December 15, 2021
Suppress memorysanitizer for code that calls LAPACK kernels.
Authored January 7, 2022
[jax2tf] Add jax2tf_associative_scan_reductions flag
Authored January 13, 2022
NotImplementedError for new_checkpoint-of-xmap
Authored January 20, 2022
jnp.[nan]argmin/max: implement keepdims
Authored January 20, 2022
fix jaxpr type checking error messages
Authored January 9, 2022
Change primitive arguments to parallel and sparse primitives to make parameters hashable.
Authored December 16, 2021
Fix test failures under Numpy 1.22.
Authored January 4, 2022
Add division by zero check.
Authored December 20, 2021
Add GDA to the API pages
Authored January 21, 2022

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.76

hawkinsp
mattjj
jakevdp
froystig
yashk2810
gnecula
LenaMartens
a-googler
tomhennigan
Edenhofer

Directory Browser for jaxlib-v0.1.76

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
    • Includes precompiled SASS for NVidia compute capability 8.0 GPUS (e.g. A100). Removes precompiled SASS for compute capability 6.1 so as not to increase the number of compute capabilities: GPUs with compute capability 6.1 can use the 6.0 SASS.
    • With jaxlib 0.1.76, JAX uses the MHLO MLIR dialect as its primary target compiler IR by default.
  • Breaking changes
    • Support for NumPy 1.18 has been dropped, per the deprecation policy. Please upgrade to a supported NumPy version.
  • Bug fixes
    • Fixed a bug where apparently identical pytreedef objects constructed by different routes do not compare as equal (#9066).
    • The JAX jit cache requires two static arguments to have identical types for a cache hit (#9311).
GitClear uses cookies to ensure you get the best experience on our website. Learn more