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

Jax: jax-v0.4.3-rc Release

Release date:
February 8, 2023
Previous version:
jax-v0.4.2 (released January 25, 2023)
Magnitude:
8,565 Diff Delta
Contributors:
23 total committers
Data confidence:
Commits:
106 impacting commits

106 Commits in this Release

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

Bump minimum jaxlib_version to 0.4.2 i.e xla_extension_version == 119 and mlir_api_version == 43
Authored February 6, 2023
migrate internal dependencies from `jax.interpreters.ad` to `jax._src.interpreters.ad`
Authored February 7, 2023
Clean up: remove unused helper functions
Authored February 1, 2023
[sparse] support BCSR in sparsify transform
Authored February 3, 2023
Replace uses of deprecated JAX sharding APIs with their new names in jax.sharding.
Authored February 3, 2023
Make pjit's cache global just like `jit`'s cache. This will allow cache hits in C++ when `pjit(f)(jnp.arange(3.))` is executed twice.
Authored February 7, 2023
Remove the usage of `_arrays` from tests
Authored January 31, 2023
[shape_poly] Fix the hashing and equality of symbolic dimensions
Authored February 4, 2023
Fix the warning being raised when jax.Array is True about using jax.Array
Authored January 27, 2023
Add support for a list of `allow_spmd_sharding_propagation_to_output`. This gives us more flexibility to tell SPMD which shardings to override.
Authored February 4, 2023
Add static_argnums to custom_partitioning.
Authored February 3, 2023
[jax2tf] port sharding tests to use MHLO
Authored January 27, 2023
Copy the jit docs and paste it inside the new jit fork.
Authored February 4, 2023
Remove placeholder functions for unimplemented NumPy functions.
Authored February 2, 2023
[sparse] add BCSR.to_bcoo and from_bcoo methods
Authored January 30, 2023
Added scipy.stats.rankdata
Authored February 7, 2023
[sparse] implement __len__ on sparse objects
Authored February 1, 2023
[shape_poly] Add support for evaluating div/mod for DimExpr
Authored February 3, 2023
[sparse] support all unbatched 1D convolutions
Authored February 3, 2023
Add a .cost_analysis() on lowered but uncompiled computations.
Authored February 6, 2023
Delete GDA serialization tests
Authored January 28, 2023
Add option to run tests with persistent compilation cache enabled.
Authored February 7, 2023
remove opaque dtype aval translation to MLIR types
Authored January 27, 2023
Revert "Merge pull request #14113 from botev:main"
Authored February 2, 2023
Move jax.interpreters.xla to jax._src.interpreters.xla.
Authored February 7, 2023
minor tweaks to type annotations, specialize code on those types
Authored February 3, 2023
[sparse] implement sparse rule for lax.rev
Authored February 1, 2023
Adding @custom_partitioning to jax.experimental API doc
Authored January 19, 2023
checkify: cache jaxpr formation so we don't always retrace
Authored February 1, 2023
[sparse] add support for simple 1D convolutions
Authored February 2, 2023
DOC: update contributing guide
Authored February 7, 2023
Update signature of ad.defbilinear to simplify transpose rules
Authored January 31, 2023
Add `axis_substitution_rules` rule for pmap so that pjit(pmap) with an axis_index works properly
Authored January 26, 2023
[ROCm] Re-enable some linalg and sparse tests
Authored February 7, 2023
Begin deprecation of public jax.ShapedArray
Authored January 30, 2023
Sharp bits: mention alternatives to lax.cond
Authored February 2, 2023
Removing HLO dump from docstring, using assert(re.search(...))
Authored January 23, 2023
Checkify: Validate format arguments to make sure they're arrays
Authored January 24, 2023
[shard_map] add rep rule for axis_index, trivial test
Authored February 6, 2023
Add environment variable check in additional to libtpu check for cloud tpu vm
Authored January 25, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.6-rc Released March 9, 2023
5,714 Ξ”
jax-v0.4.5 Released March 2, 2023
11,708 Ξ”
jax-v0.4.4 Released February 16, 2023
0 Ξ”
jax-v0.4.4-rc Released February 16, 2023
6,660 Ξ”
jax-v0.4.3 Released February 8, 2023
0 Ξ”
jax-v0.4.3-rc Released February 8, 2023
8,565 Ξ”
jax-v0.4.2 Released January 25, 2023
0 Ξ”
jax-v0.4.2-rc Released January 25, 2023
26,261 Ξ”
jax-v0.4.1 Released December 13, 2022
0 Ξ”
jax-v0.4.1-rc Released December 13, 2022
592 Ξ”

Top Contributors in jax-v0.4.3-rc

yashk2810
jakevdp
hawkinsp
froystig
gnecula
mattjj
skye
pschuh
nvlcambier
a-googler

Directory Browser for jax-v0.4.3-rc

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