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,168 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'
Authored February 9, 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
Add experimental support for SPMD lowering of xmap via MANUAL sharding annotations
Authored February 4, 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
Added an explicit Any return type to lax_numpy.ndarray methods
Authored February 8, 2022
add and test pytree utils for better errors
Authored January 28, 2022
Add `lax.conv_general_dilated_local`
Authored May 13, 2021
[checkify] handle custom_jvp
Authored February 9, 2022
[JAX] Move the ann recall computation to ann.py
Authored February 2, 2022
Speed up source location computation when lowering a jaxpr to HLO/MHLO
Authored February 9, 2022
Apply suggestions from code review
Authored February 8, 2022
Use thread local positional semantics and change thread_resources to subclass from threading.local
Authored February 9, 2022
[checkify] add custom_vjp support
Authored February 9, 2022
[mhlo] jax infeed/outfeed lowering changes in response to detupling of the ops
Authored February 2, 2022
Ensure that tree_util.Partial's .func attribute is stable
Authored February 3, 2022
Replace core.pp_eqn_compact() with core.str_eqn_compact()
Authored February 9, 2022
tests: add unsigned ints to all_dtypes
Authored February 7, 2022
Support string fft_type values in lax.fft
Authored February 2, 2022
[JAX] Update the ANN document
Authored February 5, 2022
Strip named_shape and weak_type from aval when donating buffers
Authored February 10, 2022
Fix build and address suggestion
Authored February 8, 2022
NumpyUfuncTests: speed up test generation
Authored January 31, 2022
Expand implementation of lax_numpy._quantile to allow the input of a tuple as axis argument
Authored February 6, 2022
Allow nesting MANUAL-style xmaps in pjits
Authored February 7, 2022
Add batch_axis to variance scaling initializers
Authored February 5, 2022
Catch the device buffer order with the expected global_mesh.local_devices order that we should get
Authored February 8, 2022
jnp.split: push inputs to device before splitting
Authored February 4, 2022
Add a replica_id check so that the checkpoint inconsistency that was observed due to wrong replica_id calculation will not happen again
Authored February 3, 2022
[mhlo] Phase I: Remove tuples from mhlo Infeed/Outfeed/Send/Recv Ops
Authored February 2, 2022
Handle serialization of arrays with shape `(0,)`. 0
Authored February 3, 2022
Release jax and jaxlib 0.3.0 as per the new release process
Authored February 10, 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
[jax2tf] minor changes to the documentation for TF Lite
Authored February 1, 2022
CI: error if docstring rewrite fails
Authored February 7, 2022
[sparse] fix bcoo_extract batching rule
Authored February 3, 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.5 Released April 7, 2022
8,706 Ξ”
jax-v0.3.4 Released March 18, 2022
271 Ξ”
jax-v0.3.3 Released March 17, 2022
609 Ξ”
jax-v0.3.2 Released March 16, 2022
13,073 Ξ”
jax-v0.3.1 Released February 18, 2022
6,870 Ξ”
jax-v0.3.0 Released February 10, 2022
5,168 Ξ”
jax-v0.2.28 Released February 2, 2022
774 Ξ”
jax-v0.2.27 Released January 18, 2022
5,555 Ξ”
jax-v0.2.26 Released December 8, 2021
1,351 Ξ”
jax-v0.2.25 Released October 25, 2021
2,993 Ξ”

Top Contributors in jax-v0.3.0

hawkinsp
LenaMartens
mattjj
apaszke
yashk2810
jakevdp
tlu7
superbobry
romanngg
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 about our cookie policy