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

Jax: jaxlib-v0.4.9 Release

Release date:
May 9, 2023
Previous version:
jaxlib-v0.4.7 (released March 27, 2023)
Magnitude:
25,274 Diff Delta
Contributors:
43 total committers
Data confidence:
Commits:
380 impacting commits

380 Commits in this Release

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

Add initial implementation of a `run_state` primitive
Authored February 17, 2023
Add documentation for callback functions
Authored April 7, 2023
[shape_poly] Improved computation of dimension variables for native serialization
Authored March 29, 2023
[shard-map] fix eager shmap+prngs, revise phys aval/sharding logic
Authored April 5, 2023
[shape_poly] Refactor the unification of the argument abstract values with the actual arguments
Authored April 26, 2023
[shape_poly] Cleanup handling of dimension variables.
Authored March 29, 2023
Move jax.interpreters.partial_eval to jax._src.interpreters.partial_eval.
Authored March 27, 2023
Bump minimum jaxlib version from 0.4.6 to 0.4.7.
Authored March 28, 2023
Rename jax._src.sharding_utils to jax._src.op_shardings.
Authored April 6, 2023
[jax2tfO] Add support for calling an exported JAX function
Authored April 20, 2023
Split ShardingSpecs and most of the helpers for constructing them into a separate file (jax/_src/sharding_specs.py).
Authored April 6, 2023
Change np.prod->math.prod
Authored April 13, 2023
[shape_poly] Lowering sharding annotations in presence of dynamic shapes
Authored April 5, 2023
Add a C++ implementation of safe map.
Authored April 11, 2023
Preserve shardings on the output of pjit that were provided on the arguments.
Authored April 9, 2023
Add test framework for module attribute
Authored April 21, 2023
Delete some dead code that pertained to sharded_jit.
Authored April 12, 2023
[2/n] Embed the tf.Graph into the stablehlo.custom_call.
Authored April 26, 2023
[call_tf] Some cleanup of call_tf
Authored April 27, 2023
[shape_poly] Refactor the computation of the dimension variables in native serialization
Authored March 12, 2023
Change to simpler import for jax.config
Authored April 21, 2023
Update the CUDA installation instructions.
Authored March 29, 2023
KeyArray: remove _stackable registration mechanism
Authored April 24, 2023
[djax] handle simple reshapes and size-0 checks
Authored April 20, 2023
random.KeyArray: more comprehensive operator tests
Authored April 24, 2023
Fully implement and test axes argument to jax.scipy.signal.fftconvolve
Authored April 12, 2023
[jax2tf] Simplify back_compat_test.py to use jax_export mechanisms to run
Authored April 29, 2023
always lower all_to_all to AllToAll
Authored April 12, 2023
Add a C++ safe_zip implementation.
Authored April 11, 2023
[shape_poly] Improve testing of vmap test harnesses
Authored May 9, 2023
Use ml_dtypes definition for jnp.finfo
Authored May 4, 2023
add `jax.random.bits`
Authored May 2, 2023
Add more return type annotations.
Authored April 12, 2023
Add implementation of jax.scipy.fftconvolve
Authored April 7, 2023
Remove global_arg_shapes from pmap since it was only used for sharded_jit and sharded_jit was removed from JAX a long time ago
Authored March 29, 2023
[shape_poly] Extend the handling of jnp.arange with shape polymorphism.
Authored March 27, 2023
Remove `experimental_cpp_jit` since that flag is unused and also remove `experimental_cpp_pjit`.
Authored April 7, 2023
internal: remove aliasing of builtins from lax_numpy
Authored April 21, 2023
Small cleanups to pxla.py.
Authored April 21, 2023
Remove tupling of custom call results.
Authored May 1, 2023

Browse Other Releases

Latest Pending Unreleased 😎
0.3.23 Released October 12, 2022
305 Ξ”
0.3.22 Released October 11, 2022
11,637 Ξ”
0.3.21 Released September 30, 2022
1,561 Ξ”
0.3.20 Released May 11, 2023
150,746 Ξ”
jaxlib-v0.4.10 Released May 11, 2023
1,233 Ξ”
jaxlib-v0.4.9 Released May 9, 2023
25,274 Ξ”
jaxlib-v0.4.7 Released March 27, 2023
19,033 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
17,943 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
6,717 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
7,661 Ξ”

Top Contributors in jaxlib-v0.4.9

jakevdp
yashk2810
hawkinsp
gnecula
mattjj
sharadmv
froystig
skye
maxwillzq
a-googler

Directory Browser for jaxlib-v0.4.9

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