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

Jax: jaxlib-v0.3.0 Release

Release date:
February 10, 2022
Previous version:
jaxlib-v0.1.76 (released January 27, 2022)
Magnitude:
8,876 Diff Delta
Contributors:
21 total committers
Data confidence:
Commits:
97 impacting commits

97 Commits in this Release

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

djax: let make_jaxpr build dyn shape jaxprs
Authored January 21, 2022
Introduce a new jax/jaxlib versioning scheme.
Authored February 2, 2022
Change lax.select_p to be an n-ary predicate, 'lax.select_n_p'. Change `lax.select()` to be a thin shim around the new n-ary version.
Authored February 9, 2022
Add experimental support for SPMD lowering of xmap via MANUAL sharding annotations
Authored February 4, 2022
Checkify: rename some symbols and add some docstrings.
Authored January 10, 2022
Rollback of: Implement the JAX transfer guard API
Authored February 9, 2022
[JAX] Added jit-able singular value decomposition.
Authored February 3, 2022
[JAX] Re-enables lowering bcoo dot general to cuSparse.
Authored January 31, 2022
Checkify: Fix docstring formatting and polish enabled_errors sets.
Authored February 10, 2022
disable implicit rank promotion in a number of remaining tests
Authored January 28, 2022
add and test pytree utils for better errors
Authored January 28, 2022
Add `lax.conv_general_dilated_local`
Authored May 13, 2021
Added an explicit Any return type to lax_numpy.ndarray methods
Authored February 8, 2022
fix caching example in jit tutorial
Authored January 25, 2022
[XLA:TPU] Add 'aggregate_to_topk' option to ann in jax
Authored January 31, 2022
[checkify] handle custom_jvp
Authored February 9, 2022
Couple of changes because of the serialization inconsistencies being observed.
Authored February 1, 2022
Speed up source location computation when lowering a jaxpr to HLO/MHLO.
Authored February 9, 2022
tests: add unsigned ints to all_dtypes
Authored February 7, 2022
[checkify] add custom_vjp support
Authored February 9, 2022
Clarify the required NVidia driver versions in the JAX readme instructions.
Authored January 28, 2022
Add batch_axis to variance scaling initializers
Authored February 5, 2022
Apply suggestions from code review
Authored February 8, 2022
[JAX] Move the ann recall computation to ann.py.
Authored February 2, 2022
Use thread local positional semantics and change thread_resources to subclass from threading.local.
Authored February 9, 2022
Allow nesting MANUAL-style xmaps in pjits
Authored February 7, 2022
introduce `custom_batching.sequential_vmap`
Authored January 16, 2022
Add the cache back now that Mesh's __hash__ is also being hashed on `self.devices.shape`.
Authored February 1, 2022
Ensure that tree_util.Partial's .func attribute is stable.
Authored February 3, 2022
[mhlo] jax infeed/outfeed lowering changes in response to detupling of the ops.
Authored February 2, 2022
Support string fft_type values in lax.fft.
Authored February 2, 2022
Replace core.pp_eqn_compact() with core.str_eqn_compact().
Authored February 9, 2022
Strip named_shape and weak_type from aval when donating buffers.
Authored February 10, 2022
Strip named_shape and weak_type from aval when donating buffers.
Authored January 28, 2022
Expand implementation of lax_numpy._quantile to allow the input of a tuple as axis argument
Authored February 6, 2022
Catch the device buffer order with the expected global_mesh.local_devices order that we should get. This is to make sure that we don't get some cryptic message from XLA and catch it in GDA itself.
Authored February 8, 2022
NumpyUfuncTests: speed up test generation
Authored January 31, 2022
Strip named_shape and weak_type from aval when donating buffers.
Authored January 28, 2022
[JAX] Update the ANN document.
Authored February 5, 2022
maint: update pre-commit package versions & fix new mypy errors
Authored January 31, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.3.10 Released May 4, 2022
13,336 Ξ”
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 Ξ”

Top Contributors in jaxlib-v0.3.0

hawkinsp
mattjj
LenaMartens
yashk2810
jakevdp
tlu7
apaszke
a-googler
romanngg
superbobry

Directory Browser for jaxlib-v0.3.0

We haven't yet finished calculating and confirming the files and directories changed in this release. Please check back soon.

Release Notes Published

  • Changes
    • Bazel 5.0.0 is now required to build jaxlib.
    • jaxlib version has been bumped to 0.3.0. Please see the design doc for the explanation.
GitClear uses cookies to ensure you get the best experience on our website. Learn more