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

Jax: jax-v0.3.0 Release

Release date:
February 10, 2022
Previous version:
jax-v0.2.28 (released February 2, 2022)
Magnitude:
5,978 Diff Delta
Contributors:
17 total committers
Data confidence:
Commits:
68 impacting commits

68 Commits in this Release

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

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
Checkify: Fix docstring formatting and polish enabled_errors sets.
Authored February 10, 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
[checkify] handle custom_jvp
Authored February 9, 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
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
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
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
[JAX] Update the ANN document.
Authored February 5, 2022
Fix build and address suggestion
Authored February 8, 2022
Handle serialization of arrays with shape `(0,)`. These arrays are usually empty lists (`np.array([])`)
Authored February 3, 2022
Add a replica_id check so that the checkpoint inconsistency that was observed due to wrong replica_id calculation will not happen again. We will catch this at the GDA level in `__init__` and raise an error before going to checkpointing.
Authored February 3, 2022
[sparse] fix bcoo_extract batching rule
Authored February 3, 2022
jnp.split: push inputs to device before splitting
Authored February 4, 2022
Release jax and jaxlib 0.3.0 as per the new release process.
Authored February 10, 2022
[jax2tf] minor changes to the documentation for TF Lite
Authored February 1, 2022
[mhlo] Phase I: Remove tuples from mhlo Infeed/Outfeed/Send/Recv Ops.
Authored February 2, 2022
Work around call expansion in MHLO->HLO lowering.
Authored February 2, 2022
Add complex number DLPack support to JAX and TensorFlow.
Authored February 9, 2022
Add test and jet-primitive for dynamic_slice
Authored February 8, 2022
Fix missing handler when lexically capturing a ShardedDeviceArray when MLIR enabled.
Authored February 8, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.4 Released March 18, 2022
331 Ξ”
jax-v0.3.3 Released March 17, 2022
679 Ξ”
jax-v0.3.2 Released March 16, 2022
13,751 Ξ”
jax-v0.3.1 Released February 18, 2022
6,721 Ξ”
jax-v0.3.0 Released February 10, 2022
5,978 Ξ”
jax-v0.2.28 Released February 2, 2022
829 Ξ”
jax-v0.2.27 Released January 18, 2022
6,502 Ξ”
jax-v0.2.26 Released December 8, 2021
1,528 Ξ”
jax-v0.2.25 Released October 25, 2021
2,029 Ξ”
jax-v0.2.24 Released October 19, 2021
298 Ξ”

Top Contributors in jax-v0.3.0

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

Directory Browser for jax-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
    • jax 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