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

Jax: jax-v0.3.2 Release

Release date:
March 16, 2022
Previous version:
jax-v0.3.1 (released February 18, 2022)
Magnitude:
13,073 Diff Delta
Contributors:
24 total committers
Data confidence:
Commits:
150 impacting commits

150 Commits in this Release

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

Update JAX to use new math libraries in ROCm-5.0
Authored February 15, 2022
[sparse] refactor internal implementation of sparsify transform
Authored March 7, 2022
Add separate mechanism for threading name stacks to the lowering
Authored October 28, 2021
[sparse] change call signature of coo primitive wrappers
Authored March 7, 2022
re-implement `custom_transpose` without upfront staging
Authored January 21, 2022
Increase minimum jaxlib version to 0.3.0
Authored March 4, 2022
Document jax.nn.initializers
Authored March 7, 2022
Rollback of:
Authored March 2, 2022
lax_numpy.py: factor ufuncs into their own private submodule
Authored March 15, 2022
remove `_const` from public `jax.lax` module
Authored March 7, 2022
Set in_positional_semantics should be GLOBAL for fully replicated values to avoid recompilation
Authored March 15, 2022
[sparse] track sorted columns for COO GPU lowerings
Authored March 9, 2022
Factor-out pieces of lax_numpy.py
Authored March 1, 2022
Make `mesh_axes` on GDA strict by only allowing PartitionSpecs to be consistent with pjit
Authored March 7, 2022
Refactor Jaxpr pretty-printing to use a `JaxprPpSettings` named tuple
Authored March 9, 2022
Remove jax.ops.index... functions
Authored February 17, 2022
[JAX] Update ann to use XLA based fallback ApproxTopK
Authored March 15, 2022
Documentation improvements
Authored March 8, 2022
remove `_convert_element_type` from public `jax.lax` module
Authored March 10, 2022
lax_numpy.py: factor out some common utilities
Authored March 10, 2022
cleanup: remove _constant_like in favor of lax._const
Authored March 2, 2022
[sparse]: COO: check for sorted rows before cusparse lowering
Authored March 8, 2022
convert to doctest
Authored March 3, 2022
Always treat all mesh axes controlled by xmap as MANUAL
Authored February 22, 2022
Add the input avals to Lowered and Compiled
Authored March 9, 2022
Add an API compatibility policy
Authored February 17, 2022
Index update operators: add scatter_apply()
Authored February 18, 2022
Use real Precision type for lax.PrecisionType
Authored March 4, 2022
Support gamma distribution with PRNGKeys other than threefry2x32
Authored March 10, 2022
Improve TPU v2 and v3 mesh_utils.create_device_mesh logic
Authored March 8, 2022
Handle jaxpr constants correctly in MLIR lowering of conditional branches
Authored March 1, 2022
Checkify: address initial feedback
Authored February 17, 2022
format nicely
Authored March 4, 2022
improvements to autodidax reduce_sum
Authored March 16, 2022
Use sharded shape to compute aliasing
Authored March 10, 2022
in tests, compare jnp operations on PRNGKeyArrays to the same on jnp arrays
Authored February 17, 2022
Integrate LLVM at llvm/llvm-project@eb27da7dec67
Authored March 3, 2022
[JAX] Update comments and documents for ANN
Authored February 24, 2022
Adds use_auto_spmd_partitioning and propagates its value down the stack
Authored March 15, 2022
custom_prng: better error messages for key validation
Authored March 4, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.6 Released April 13, 2022
5,312 Ξ”
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 Ξ”

Top Contributors in jax-v0.3.2

jakevdp
reza-amd
hawkinsp
yashk2810
froystig
sharadmv
a-googler
jblespiau
mattjj
LenaMartens

Directory Browser for jax-v0.3.2

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