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

Jax: jax-v0.4.0-rc Release

Release date:
December 8, 2022
Previous version:
jax-v0.3.25 (released November 15, 2022)
Magnitude:
15,791 Diff Delta
Contributors:
31 total committers
Data confidence:
Commits:
177 impacting commits

177 Commits in this Release

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

Rewrite Checkify to support tracking different error types
Authored October 18, 2022
jax.Array tutorial
Authored November 16, 2022
[typing] annotate jax.experimental.sparse
Authored December 5, 2022
Rollback of "Add CUDNN custom call for LSTM. 0
Authored November 23, 2022
Switch to new-style f-strings
Authored December 1, 2022
[sparse] improve testing framework
Authored November 16, 2022
Add CUDNN custom call for LSTM. 0
Authored November 28, 2022
[JAX] Opt into high precision matrix multiplications in JAX tests that fail on A100
Authored November 15, 2022
Remove local_imports of sharding.py. 0
Authored November 30, 2022
Refactor create_cpp_call to be a method on MeshExecutable
Authored November 18, 2022
[sparse] refactor with sparse.test_util
Authored November 16, 2022
Remove typing_extensions dependency
Authored December 5, 2022
Use jax.Array by default for doctests
Authored November 15, 2022
Add JAX monitoring library that instruments code via events
Authored November 15, 2022
Remove local imports of array.py. 0
Authored November 30, 2022
Remove unused potrf kernels
Authored November 17, 2022
[call_tf] Use the same platform for TF lowering as the embedding JAX computation
Authored December 1, 2022
Cache the exec time `tf.get_current_name_scope` into `_thread_local_state.exec_time_tf_name_scope` and add it as prefix during tracing
Authored December 6, 2022
Transpose all_gather to reduce_scatter
Authored November 15, 2022
Move `PartitionSpec` and `Mesh` out of experimental and into the `sharding` namespace
Authored December 2, 2022
describe partitionable RNG mode in parallelism doc
Authored November 29, 2022
Allow compiling and then serializing jax.stages.Lowered
Authored November 16, 2022
Support AOT serialization of pmap
Authored November 23, 2022
[x64] minor test-only updates for better type safety
Authored November 30, 2022
Added more accurate version of the betaln function
Authored November 17, 2022
Don't call `dtypes.result_type()` unnecessarily on the type of an array during abstractification
Authored November 30, 2022
[x64] minor fixes for lax_numpy_test type safety
Authored December 1, 2022
[x64] make tests more type-safe
Authored December 2, 2022
[sparse] support dense dimensions in bcoo_reshape
Authored November 14, 2022
Optimize `host_local_array_to_global_array` by caching the local to global conversion and flattening of axis resources
Authored November 21, 2022
Pickling of Sharding classes: use module level functions when deserializing
Authored November 22, 2022
[x64] update api_test for type safety
Authored November 29, 2022
Make `jnp.copy` work with all shardings especially `PmapSharding`
Authored December 7, 2022
Do not mirror JAX config options back to ABSL flags
Authored December 2, 2022
Mark arguments to ufuncs as positional-only
Authored December 6, 2022
custom vmap: support closure and staged constants
Authored November 23, 2022
Remove opt-barrier fallbacks
Authored November 17, 2022
Remove redundant dtype canonicalization from jax.device_put()
Authored November 29, 2022
[jax2tf] Upgrade examples to use optax instead of flax.optim
Authored November 21, 2022
[TPU] Add cutoff for nearly diagonal matrices in QDWH-eigh algorithm
Authored November 23, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.1 Released December 13, 2022
0 Ξ”
jax-v0.4.1-rc Released December 13, 2022
535 Ξ”
jax-v0.4.0 Released December 12, 2022
0 Ξ”
jax-v0.4.0-rc1 Released December 12, 2022
2,479 Ξ”
jax-v0.4.0-rc Released December 8, 2022
15,791 Ξ”
jax-v0.3.25 Released November 15, 2022
0 Ξ”
jax-v0.3.25-rc3 Released November 15, 2022
28 Ξ”
jax-v0.3.25-rc2 Released November 15, 2022
3 Ξ”
jax-v0.3.25-rc1 Released November 15, 2022
3 Ξ”
jax-v0.3.25-rc Released November 15, 2022
8,747 Ξ”

Top Contributors in jax-v0.4.0-rc

jakevdp
yashk2810
LenaMartens
a-googler
hawkinsp
pschuh
froystig
zhangqiaorjc
gnecula
apaszke

Directory Browser for jax-v0.4.0-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 about our cookie policy