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:
8,813 Diff Delta
Contributors:
39 total committers
Data confidence:
Commits:
148 impacting commits

148 Commits in this Release

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

[MHLO] Add direct MHLO lowerings for most linear algebra kernels
Authored April 6, 2022
[sparse] Change call signature of bcoo primitive wrappers
Authored March 29, 2022
First attempt to enable auto-sharding. 0
Authored April 1, 2022
Fix for hipsparse in ROCm
Authored March 17, 2022
Checkify: add way to embed runtime info in error messages
Authored March 8, 2022
add random.loggamma and improve dirichlet & beta implementation
Authored March 21, 2022
Factor-out reductions from lax_numpy.py
Authored March 18, 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
Checkify: add axis and axis size to OOB error message
Authored March 30, 2022
Speed up GDA initialization by making local_shards lazy and hiding checks behind `config.jax_enable_checks` flag
Authored March 29, 2022
Remove the jax_enable_mlir flag. 0
Authored April 4, 2022
[MHLO] Prefer backend-specific HLO lowerings instead of non-backend-specific MHLO lowerings
Authored April 6, 2022
allow for recursive uses of `custom_transpose`
Authored March 25, 2022
[jax2tf] Updates `custom_assert` for jax2tf SVD (primitive) limitations
Authored March 30, 2022
require consistent output structure in custom vmap rules
Authored January 28, 2022
factor AOT types out to a `stages` module
Authored March 15, 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
Deprecate jax.tree_util.tree_multimap
Authored April 1, 2022
improve remat transpose caching (cf. #9661)
Authored March 25, 2022
Migrate 'jaxlib' CPU custom-calls to the status-returning API
Authored March 30, 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
[mhlo] Clean up ops that can use InferTensorTypeWithReify
Authored March 30, 2022
Add weakreaf_lru_cache to prevent caches from pinning jaxprs
Authored March 19, 2022
[MHLO] Add MHLO lowering for PRNG kernels
Authored April 6, 2022
refine occurrences of the "forward-mode disallowed" error under new `custom_vjp`
Authored March 26, 2022
[sparse] bcoo_mul: support mixing batch & sparse dims
Authored March 18, 2022
[sparse] add bcoo_add_batchdim
Authored April 6, 2022
improvements to autodidax reduce_sum
Authored March 16, 2022
random_test: add tests of random values for distributions
Authored March 23, 2022
Expose AutoSharding's mesh_shape and mesh_ids options to JAX
Authored April 1, 2022
Implement `jnp.trace` in terms of `jnp.diagonal`
Authored April 6, 2022
Revert: implement `jnp.trace` in terms of `jnp.diagonal`
Authored April 6, 2022
doc: add examples of using partial with jit
Authored March 29, 2022
avoid batching units in cond partial eval
Authored March 23, 2022
Avoid retracing when a host_callback.call is called multiple times with the same function
Authored April 1, 2022
[shape_poly] Change internal representation of parametric shapes
Authored March 28, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.3.15 Released July 22, 2022
290 Ξ”
jaxlib-v0.3.14 Released June 27, 2022
25 Ξ”
jaxlib-v0.3.10 Released May 4, 2022
0 Ξ”
jaxlib-v0.3.7 Released April 15, 2022
6,532 Ξ”
jaxlib-v0.3.5 Released April 7, 2022
8,813 Ξ”
jaxlib-v0.3.2 Released March 15, 2022
19,111 Ξ”
jaxlib-v0.3.0 Released February 10, 2022
5,413 Ξ”
jaxlib-v0.1.76 Released January 27, 2022
6,035 Ξ”
jaxlib-v0.1.75 Released December 7, 2021
39 Ξ”
jaxlib-v0.1.74 Released November 17, 2021
5,984 Ξ”

Top Contributors in jaxlib-v0.3.5

jakevdp
hawkinsp
froystig
yashk2810
tlu7
LenaMartens
reza-amd
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 about our cookie policy