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:
7,011 Diff Delta
Contributors:
19 total committers
Data confidence:
Commits:
67 impacting commits

67 Commits in this Release

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

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
fix a custom_vjp post_process bug, related cleanups
Authored December 11, 2021
[sparse] generalize metadata argument in BCOO primitives
Authored January 7, 2022
PartitionSpec: allow partially specified sharding
Authored January 13, 2022
Optimizations for GDA to make creating GDA faster.
Authored January 13, 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
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
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
Suppress memorysanitizer for code that calls LAPACK kernels.
Authored January 7, 2022
[jax2tf] Add jax2tf_associative_scan_reductions flag
Authored January 13, 2022
fix jaxpr type checking error messages
Authored January 9, 2022
Add division by zero check.
Authored December 20, 2021
custom batching jvp tests
Authored December 31, 2021
JVP rule for custom_vmap
Authored December 3, 2021
Re-applying #9136 after it was rolled back.
Authored January 9, 2022
remove axis name logic from Primitive / bind
Authored January 5, 2022
checkify: tweak some organization and names
Authored January 11, 2022
Resolves issue 8744
Authored December 7, 2021
searchsorted: use correct ordering for complex inputs
Authored January 13, 2022
Implement InferTypeOpInterface on TransposeOp (NFC)
Authored January 5, 2022
Add scatter OOB error.
Authored December 23, 2021
Clean up backwards-compatibility support for remat lowering
Authored January 4, 2022
jnp.unique: properly handle NaN values
Authored January 13, 2022
Add missing annotations to lax
Authored December 10, 2021
Copybara import of the project:
Authored January 11, 2022
[JAX] xmap: specify shardings only on the named dimensions in the weak mode.
Authored January 7, 2022
Fix a bug that promoted t to a complex in odeint, and modify a test so it would have caught it
Authored December 1, 2021
clean up WrappedFun.call_wrapped refs on exception
Authored January 5, 2022
add `random.default_prng_impl` to retrieve the default PRNG implementation
Authored January 13, 2022
cache initial-style jaxpr transformations
Authored January 14, 2022
[mhlo] Remove tuples from mhlo If/Case op.
Authored January 9, 2022
Lift non-contiguous mesh restriction for fully replicated values
Authored January 8, 2022

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
jaxlib-v0.1.76 Released January 27, 2022
7,011 Ξ”
jaxlib-v0.1.75 Released December 7, 2021
39 Ξ”
jaxlib-v0.1.74 Released November 17, 2021
5,231 Ξ”
jaxlib-v0.1.73 Released October 18, 2021
0 Ξ”
jaxlib-v0.1.72 Released October 12, 2021
11,692 Ξ”

Top Contributors in jaxlib-v0.1.76

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

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