Home » Oreps » Jax » All Releases » jaxlib-v0.4.3 Release
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Watch
0
Commit Activity
Directory Browser
Velocity History
Releasesjaxlib-v0.4.3
Basic Details Issues Addressed Top Contributors Directory Browser
Jax jaxlib-v0.4.3

Jax: jaxlib-v0.4.3 Release

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

105 Commits in this Release

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

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 static_argnums to custom_partitioning.
Authored February 3, 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
[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
Use pareto optimal step size for computing numerical Jacobians in JAX. This allows us to tighten the tolerances in gradient unit testing significantly, especially for float64 and complex128.
Authored January 25, 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
minor tweaks to type annotations, specialize code on those types
Authored February 3, 2023
Move jax.interpreters.xla to jax._src.interpreters.xla.
Authored February 7, 2023
[sparse] implement sparse rule for lax.rev
Authored February 1, 2023
checkify: cache jaxpr formation so we don't always retrace
Authored February 1, 2023
Adding @custom_partitioning to jax.experimental API doc
Authored January 19, 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
Removing HLO dump from docstring, using assert(re.search(...))
Authored January 23, 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
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 😎
0.3.21 Released September 30, 2022
1,561 Ξ”
0.3.20 Released March 9, 2023
101,633 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
17,010 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
6,724 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
7,645 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
24,737 Ξ”
jaxlib-v0.4.1 Released December 13, 2022
596 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
16,803 Ξ”
jaxlib-v0.3.25 Released November 15, 2022
6,881 Ξ”
jaxlib-v0.3.24 Released November 4, 2022
23,948 Ξ”

Top Contributors in jaxlib-v0.4.3

jake-vanderplas-1c1d
yashk2810
hawkinsp
froystig
gnecula
mattjj
skye
pschuh
nvlcambier
a-googler

Directory Browser for jaxlib-v0.4.3

All files are compared to previous version, jaxlib-v0.4.2. Click here to browse diffs between other versions.

Loading File Browser...
GitClear uses cookies to ensure you get the best experience on our website. Learn more