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

Jax: jaxlib-v0.3.10 Release

Release date:
May 4, 2022
Previous version:
jaxlib-v0.3.7 (released April 15, 2022)
Magnitude:
13,335 Diff Delta
Contributors:
25 total committers
Data confidence:
Commits:
164 impacting commits

164 Commits in this Release

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

add in mlir lowering for tokens
Authored April 19, 2022
Add in runtime tokens for effectful jaxprs
Authored April 14, 2022
Increase minimum jaxlib version to 0.3.7.
Authored April 16, 2022
[mesh_utils] Support creating device meshes for hybrid networks
Authored April 18, 2022
[XLA] Call translation rule directly in xla.primitive_subcomputation.
Authored April 19, 2022
[sparse] make bcoo_sum_duplicates a primitive
Authored April 19, 2022
Add fully asynchronous checkpointing. This will allow the training to proceed forward when the checkpoint is being committed.
Authored May 3, 2022
[remove-units] prevent remat's partial eval from introducing units
Authored April 27, 2022
[sparse] Add BCOO attribute `_indices_sorted`.
Authored April 26, 2022
[remove-units] prevent scan partial eval from introducing units
Authored April 23, 2022
Add initial debug print implementation
Authored May 3, 2022
Copybara import of the project:
Authored April 26, 2022
[linalg] Add tpu svd lowering rule.
Authored April 29, 2022
Add unary xeinsum and allow named axis reductions for unary and binary xeinsums
Authored September 1, 2021
[remove-units] remove units from api_util.py
Authored April 26, 2022
[remove-units] prevent cond partial eval from introducing units
Authored April 23, 2022
[sparse] make bcoo_sort_indices a primitive
Authored April 14, 2022
[sparse] implement sparse rule for lax.concatenate_p
Authored April 28, 2022
[MHLO] Switch call_tf to use an MHLO lowering.
Authored April 19, 2022
[remove-units] avoid making xmap partial eval deal with units
Authored April 28, 2022
[linalg] Adds `full_matrices` option to TPU SVD.
Authored April 20, 2022
[remove-units] prevent ad.py from introducing units
Authored April 26, 2022
[sparse] implement sparse rule for lax.reshape_p
Authored May 2, 2022
add scan dce rule
Authored November 16, 2021
[remove-units] roll forward #10448, fix dce bug
Authored April 27, 2022
[remove-units] remove units from while partial eval
Authored April 23, 2022
jax.scipy.linalg.schur: error on 16-bit floats
Authored May 3, 2022
implement scipy.cluster.vq.vq
Authored April 20, 2022
CI: fix flake8 ignore declarations
Authored April 21, 2022
[MHLO] Add explicit XLA translation rules for primitives that lack MHLO lowerings that rely on standard_primitive registering a translation rule.
Authored April 16, 2022
[remove-units] remove units from custom_jvp/vjp
Authored April 29, 2022
[MHLO] Switch call_tf to use an MHLO lowering.
Authored April 19, 2022
[linalg] Adds `compute_uv` to TPU SVD.
Authored April 19, 2022
add partial_eval_jaxpr_nounits
Authored April 23, 2022
Make lax.linalg.qr robust to zero-dimensional inputs
Authored April 29, 2022
Attach keepalive to executable
Authored April 14, 2022
[remove-units] remove units from partial_eval.py
Authored May 2, 2022
Add CustomCall MLIR lowering for HCB `outside_call` primitive
Authored April 26, 2022
Change the default jnp.take mode to "fill".
Authored April 28, 2022
[remove-units] don't use abstract_unit for dropvar avals
Authored April 29, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.3.22 Released October 11, 2022
12,900 Ξ”
jaxlib-v0.3.20 Released September 28, 2022
55,337 Ξ”
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 Ξ”

Top Contributors in jaxlib-v0.3.10

mattjj
jakevdp
hawkinsp
sharadmv
tlu7
yashk2810
jekbradbury
YouJiacheng
anudhyan-boral-b876
a-googler

Directory Browser for jaxlib-v0.3.10

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