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

Jax: jax-v0.4.4-rc Release

Release date:
February 16, 2023
Previous version:
jax-v0.4.3 (released February 8, 2023)
Magnitude:
6,660 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
[sparse] add cusparse lowering for simplest cases of bcsr_dot_general
Authored February 9, 2023
[shape_poly] Fix some tests for shape pol with native lowering
Authored February 13, 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
[sparse] implement bcsr_broadcast_in_dim
Authored February 15, 2023
[shape_poly] Better error message for functions that do not use input arguments
Authored February 10, 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 😎
jax-v0.4.6 Released March 9, 2023
0 Ξ”
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 Ξ”

Top Contributors in jax-v0.4.4-rc

jakevdp
hawkinsp
yashk2810
gnecula
mattjj
skye
maxwillzq
jyingl3
froystig
lucashofer

Directory Browser for jax-v0.4.4-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