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

Jax: jax-v0.4.9-rc Release

Release date:
May 9, 2023
Previous version:
jax-v0.4.8 (released March 29, 2023)
Magnitude:
23,579 Diff Delta
Contributors:
43 total committers
Data confidence:
Commits:
359 impacting commits

359 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
[JAX] Improve handling of metadata in compilation cache.
Authored April 19, 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] Cleanup handling of dimension variables.
Authored March 29, 2023
Rename jax._src.sharding_utils to jax._src.op_shardings.
Authored April 6, 2023
[shape_poly] Refactor the unification of the argument abstract values with the actual arguments
Authored April 26, 2023
Split ShardingSpecs and most of the helpers for constructing them into a separate file (jax/_src/sharding_specs.py).
Authored April 6, 2023
[jax2tfO] Add support for calling an exported JAX function
Authored April 20, 2023
[shape_poly] Lowering sharding annotations in presence of dynamic shapes
Authored April 5, 2023
Change np.prod->math.prod
Authored April 13, 2023
Add test framework for module attribute
Authored April 21, 2023
Add a C++ implementation of safe map.
Authored April 11, 2023
Delete some dead code that pertained to sharded_jit.
Authored April 12, 2023
Preserve shardings on the output of pjit that were provided on the arguments.
Authored April 9, 2023
[2/n] Embed the tf.Graph into the stablehlo.custom_call.
Authored April 26, 2023
Change to simpler import for jax.config
Authored April 21, 2023
Add a C++ safe_zip implementation.
Authored April 11, 2023
KeyArray: remove _stackable registration mechanism
Authored April 24, 2023
random.KeyArray: more comprehensive operator tests
Authored April 24, 2023
[djax] handle simple reshapes and size-0 checks
Authored April 20, 2023
[call_tf] Some cleanup of call_tf
Authored April 27, 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
[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
[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
Small cleanups to pxla.py.
Authored April 21, 2023
internal: remove aliasing of builtins from lax_numpy
Authored April 21, 2023
Delete `_single_device_array_from_buf` since everything from JAX is an Array
Authored March 29, 2023
Optimize nary_dtype_rule.
Authored April 11, 2023
improve pmap in_axes/out_axes pytree prefix error messages
Authored March 29, 2023
Remove tupling of custom call results.
Authored May 1, 2023
[jax2tf] Fix backwards compatibility test to not use goldens for eigh
Authored April 12, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.11 Released May 31, 2023
0 Ξ”
jax-v0.4.11-rc Released May 31, 2023
6,799 Ξ”
jax-v0.4.10 Released May 11, 2023
0 Ξ”
jax-v0.4.10-rc Released May 11, 2023
1,233 Ξ”
jax-v0.4.9 Released May 9, 2023
1,058 Ξ”
jax-v0.4.9-rc Released May 9, 2023
23,579 Ξ”
jax-v0.4.8 Released March 29, 2023
2,231 Ξ”
jax-v0.4.7 Released March 27, 2023
0 Ξ”
jax-v0.4.7-rc1 Released March 27, 2023
0 Ξ”
jax-v0.4.7-rc Released March 27, 2023
19,399 Ξ”

Top Contributors in jax-v0.4.9-rc

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

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