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

Jax: jaxlib-v0.3.5 Release

Release date:
April 7, 2022
Previous version:
jaxlib-v0.3.2 (released March 15, 2022)
Magnitude:
9,385 Diff Delta
Contributors:
39 total committers
Data confidence:
Commits:
147 impacting commits

147 Commits in this Release

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

[sparse] Change call signature of bcoo primitive wrappers.
Authored March 29, 2022
First attempt to enable auto-sharding. This CL adds support for GDA (no SDA support yet).
Authored April 1, 2022
Checkify: add way to embed runtime info in error messages.
Authored March 8, 2022
Fix for hipsparse in ROCm.
Authored March 17, 2022
add random.loggamma and improve dirichlet & beta implementation
Authored March 21, 2022
Factor-out reductions from lax_numpy.py
Authored March 18, 2022
Checkify: add axis and axis size to OOB error message.
Authored March 30, 2022
Add istft to jax.scipy.signal.
Authored March 28, 2022
track input argument information in one tree at each AOT stage
Authored March 19, 2022
Remove the jax_enable_mlir flag. MLIR is now the only supported code path.
Authored April 4, 2022
Speed up GDA initialization by making local_shards lazy and hiding checks behind `config.jax_enable_checks` flag.
Authored March 29, 2022
allow for recursive uses of `custom_transpose`
Authored March 25, 2022
require consistent output structure in custom vmap rules
Authored January 28, 2022
[jax2tf] Updates `custom_assert` for jax2tf SVD (primitive) limitations.
Authored March 30, 2022
[MHLO] Prefer backend-specific HLO lowerings instead of non-backend-specific MHLO lowerings.
Authored April 6, 2022
Deprecate jax.tree_util.tree_multimap
Authored April 1, 2022
factor AOT types out to a `stages` module
Authored March 15, 2022
improve remat transpose caching (cf. #9661)
Authored March 25, 2022
random_test: add tests of random values for distributions
Authored March 23, 2022
protocols for the lowering and executable underlying `Lowered` and `Compiled`
Authored March 22, 2022
introduce an upgrade flag for `custom_vjp` overhaul (still experimental)
Authored March 26, 2022
lax_numpy: move poly functions into numpy.polynomial
Authored March 17, 2022
Fix race condition for weakref destructor by catching rare exceptions.
Authored March 31, 2022
* Disallow any other type other than GDA and ShapedArray for auto sharding.
Authored April 4, 2022
staging and compilation for `custom_transpose`
Authored February 18, 2022
Migrate 'jaxlib' CPU custom-calls to the status-returning API
Authored March 30, 2022
Separate jax.test_util implementations into public and private sources.
Authored April 4, 2022
Replace (deprecated) StrEnumAttr with EnumAttr.
Authored March 18, 2022
[sparse] bcoo_mul: support mixing batch & sparse dims
Authored March 18, 2022
[sparse] add bcoo_add_batchdim
Authored April 6, 2022
[mhlo] Clean up ops that can use InferTensorTypeWithReify
Authored March 30, 2022
[shape_poly] Change internal representation of parametric shapes
Authored March 28, 2022
doc: add examples of using partial with jit
Authored March 29, 2022
Add weakreaf_lru_cache to prevent caches from pinning jaxprs.
Authored March 19, 2022
When passing `ShapedArray` as input to pjit.lower() (which is not allowed normally i.e `pjit(f)(*args)`), consider it as Global. This will also help during the auto sharding change.
Authored March 24, 2022
avoid batching units in cond partial eval
Authored March 23, 2022
[MHLO] Add MHLO lowering for PRNG kernels.
Authored April 6, 2022
improvements to autodidax reduce_sum
Authored March 16, 2022
refine occurrences of the "forward-mode disallowed" error under new `custom_vjp`
Authored March 26, 2022
Avoid retracing when a host_callback.call is called multiple times with the same function.
Authored April 1, 2022

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
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 Ξ”

Top Contributors in jaxlib-v0.3.5

jakevdp
froystig
yashk2810
tlu7
LenaMartens
hawkinsp
reza-rahimi-144f
mattjj
yotarok
pschuh

Directory Browser for jaxlib-v0.3.5

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