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

Jax: jaxlib-v0.4.4 Release

Release date:
February 16, 2023
Previous version:
jaxlib-v0.4.3 (released February 8, 2023)
Magnitude:
6,724 Diff Delta
Contributors:
22 total committers
Data confidence:
Commits:
117 impacting commits

117 Commits in this Release

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

Replace uses of jnp.ndarray with jax.Array inside JAX.
Authored February 15, 2023
[sparse] implement autodiff rules for bcsr primitives
Authored February 9, 2023
simpler pretty-print for pjit, tweak custom pp rule signature
Authored February 9, 2023
Add infrastructure for managing deprecations.
Authored February 9, 2023
[jax2tf] Add shard_map tests
Authored February 10, 2023
[Rollback] Convert _arrays to return PyArray instead of PyBuffer.
Authored February 11, 2023
Plumb spmd_axis_name through batch_jaxpr2 and batch_jaxpr
Authored February 13, 2023
Add in_shardings and out_shardings argument to pjit and jit to start deprecating in_axis_resources and out_axis_resources.
Authored February 11, 2023
Add clear_cache endpoint to python pjit and cpp pjit functions.
Authored February 15, 2023
DOC: improve documentation for jax.Array methods
Authored February 14, 2023
Fix the simple bug on call_tf.replace_non_float and add unittest for floating and complex data type.
Authored February 16, 2023
[shard-map] if check_rep=False, don't call rep rules in eager
Authored February 8, 2023
[sparse] implement autodiff rules for bcsr_dot_general
Authored February 10, 2023
[PJRT C API] Register a backend factory for every PJRT plugin set in PJRT_NAMES_AND_LIBRARY_PATHS.
Authored February 9, 2023
[call_tf] Improve error reporting
Authored August 4, 2022
Implement jax.numpy.argpartition
Authored February 8, 2023
added scipy special spence
Authored February 10, 2023
[sparse] add CPU lowering rule for sparse.linalg.spsolve
Authored February 10, 2023
DOC: add better documentation for array methods
Authored February 15, 2023
Remove uses of jax.xla_computation from metadata_test.py
Authored February 16, 2023
Split PyTorch interoperability tests into their own test.
Authored February 10, 2023
jax.numpy reductions: require initial to be a scalar
Authored February 14, 2023
Remove `jax_experimental_subjaxpr_lowering_cache` since it was only for `jit` and was `False` by default. Now that jit/pjit are merged, this cache is not needed since pjit does the caching and we get it for free.
Authored February 8, 2023
Expose jax.typing & update docs
Authored February 13, 2023
[shape_poly] Fix some tests for shape pol with native lowering
Authored February 13, 2023
[sparse] add cusparse lowering for simplest cases of bcsr_dot_general
Authored February 9, 2023
[sparse] implement BCSR.__matmul__
Authored February 11, 2023
Default `jax_jit_pjit_api_merge` to True. This means that the implementation of jit and pjit have been merged but they still remain separate APIs due to the semantic difference of how they behave under the `Mesh` context manager.
Authored February 8, 2023
Point to the exact primitive name nested under jit/pjit instead of mentioning all possible ones.
Authored February 10, 2023
[shape_poly] Fix the lowering for symbolic dimension expressions for division
Authored February 16, 2023
[sparse] implement bcsr_concatenate
Authored February 15, 2023
[shard-map] add vmap spmd_axis_name support, fix vmap rule bug
Authored February 9, 2023
generalize vmap spmd_axis_name to accept tuples of axis names
Authored February 10, 2023
License cleanup.
Authored February 14, 2023
Opt into higher matmul precision for A100 and TPU tests.
Authored February 14, 2023
[shape_poly] Better error message for functions that do not use input arguments
Authored February 10, 2023
[sparse] implement bcsr_broadcast_in_dim
Authored February 15, 2023
[shard-map] add remat support, very basic test
Authored February 8, 2023
Move jax.interpreters.mlir to jax._src.interpreters.mlir.
Authored February 8, 2023
DOC: improve documentation for jax.Array methods
Authored February 14, 2023

Browse Other Releases

Latest Pending Unreleased 😎
0.3.23 Released October 12, 2022
305 Ξ”
0.3.22 Released October 11, 2022
11,644 Ξ”
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 Ξ”

Top Contributors in jaxlib-v0.4.4

jake-vanderplas-1c1d
hawkinsp
yashk2810
gnecula
mattjj
skye
maxwillzq
jyingl3
froystig
lucashofer

Directory Browser for jaxlib-v0.4.4

All files are compared to previous version, jaxlib-v0.4.3. 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