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
1
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:
5,774 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
simpler pretty-print for pjit, tweak custom pp rule signature
Authored February 9, 2023
[sparse] implement autodiff rules for bcsr primitives
Authored February 9, 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 infrastructure for managing deprecations
Authored February 9, 2023
[jax2tf] Add shard_map tests
Authored February 10, 2023
DOC: improve documentation for jax.Array methods
Authored February 14, 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
[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
DOC: add better documentation for array methods
Authored February 15, 2023
Implement jax.numpy.argpartition
Authored February 8, 2023
[sparse] add CPU lowering rule for sparse.linalg.spsolve
Authored February 10, 2023
added scipy special spence
Authored February 10, 2023
[sparse] add cusparse lowering for simplest cases of bcsr_dot_general
Authored February 9, 2023
Expose jax.typing & update docs
Authored February 13, 2023
Fix the simple bug on call_tf.replace_non_float and add unittest for floating and complex data type
Authored February 16, 2023
jax.numpy reductions: require initial to be a scalar
Authored February 14, 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
Remove `jax_experimental_subjaxpr_lowering_cache` since it was only for `jit` and was `False` by default
Authored February 8, 2023
[call_tf] Improve error reporting
Authored August 4, 2022
[sparse] implement BCSR.__matmul__
Authored February 11, 2023
Point to the exact primitive name nested under jit/pjit instead of mentioning all possible ones
Authored February 10, 2023
[shard-map] add vmap spmd_axis_name support, fix vmap rule bug
Authored February 9, 2023
Default `jax_jit_pjit_api_merge` to True. 0
Authored February 8, 2023
[shape_poly] Fix the lowering for symbolic dimension expressions for division
Authored February 16, 2023
generalize vmap spmd_axis_name to accept tuples of axis names
Authored February 10, 2023
License cleanup
Authored February 14, 2023
[sparse] implement bcsr_broadcast_in_dim
Authored February 15, 2023
DOC: improve documentation for jax.Array methods
Authored February 14, 2023
[sparse] implement bcsr_concatenate
Authored February 15, 2023
Fix pytype errors
Authored February 16, 2023
[shard-map] add remat support, very basic test
Authored February 8, 2023
Implement lowering for convolutions with dynamic padding
Authored February 14, 2023
Move jax.interpreters.mlir to jax._src.interpreters.mlir
Authored February 8, 2023
MeshComputation.cost_analysis() isn't implemented with PJRT C API
Authored February 15, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.11 Released June 3, 2023
6,363 Ξ”
jaxlib-v0.4.10 Released June 3, 2023
1,057 Ξ”
jaxlib-v0.4.9 Released June 3, 2023
23,085 Ξ”
jaxlib-v0.4.7 Released March 27, 2023
24,447 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
18,194 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
5,774 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
12,125 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
25,762 Ξ”
jaxlib-v0.4.1 Released December 13, 2022
501 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
16,767 Ξ”

Top Contributors in jaxlib-v0.4.4

jakevdp
hawkinsp
yashk2810
mattjj
gnecula
skye
jyingl3
lucashofer
froystig
maxwillzq

Directory Browser for jaxlib-v0.4.4

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