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:
14,191 Diff Delta
Contributors:
31 total committers
Data confidence:
Commits:
174 impacting commits

174 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
Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module.
Authored November 23, 2022
Rollback of "Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module."
Authored November 23, 2022
[sparse] improve testing framework
Authored November 16, 2022
Add CUDNN custom call for LSTM. Exposed as jax.experimental.rnn module.
Authored November 28, 2022
[JAX] Opt into high precision matrix multiplications in JAX tests that fail on A100.
Authored November 15, 2022
[sparse] refactor with sparse.test_util
Authored November 16, 2022
Remove local_imports of sharding.py. Adding `pxla` local imports but then cleaning those up will be super easy since those will be the only ones left and restricted to `sharding.py` file only.
Authored November 30, 2022
[call_tf] Use the same platform for TF lowering as the embedding JAX computation
Authored December 1, 2022
Transpose all_gather to reduce_scatter
Authored November 15, 2022
Refactor create_cpp_call to be a method on MeshExecutable
Authored November 18, 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
Add JAX monitoring library that instruments code via events.
Authored November 15, 2022
Remove typing_extensions dependency
Authored December 5, 2022
Use jax.Array by default for doctests
Authored November 15, 2022
describe partitionable RNG mode in parallelism doc
Authored November 29, 2022
Remove unused potrf kernels.
Authored November 17, 2022
Move `PartitionSpec` and `Mesh` out of experimental and into the `sharding` namespace. The new API endpoint is `jax.sharding.PartitionSpec` and `jax.sharding.Mesh`.
Authored December 2, 2022
[x64] minor test-only updates for better type safety
Authored November 30, 2022
Allow compiling and then serializing jax.stages.Lowered.
Authored November 16, 2022
Remove local imports of array.py. The remaining local imports are in pxla.py but I will chip away at them when we delete SDA and move some more APIs out of experimental.
Authored November 30, 2022
[x64] minor fixes for lax_numpy_test type safety
Authored December 1, 2022
[sparse] support dense dimensions in bcoo_reshape
Authored November 14, 2022
[x64] make tests more type-safe
Authored December 2, 2022
Support AOT serialization of pmap.
Authored November 23, 2022
Added more accurate version of the betaln function.
Authored November 17, 2022
[x64] update api_test for type safety
Authored November 29, 2022
Pickling of Sharding classes: use module level functions when deserializing.
Authored November 22, 2022
Don't call `dtypes.result_type()` unnecessarily on the type of an array during abstractification.
Authored November 30, 2022
Do not mirror JAX config options back to ABSL flags.
Authored December 2, 2022
custom vmap: support closure and staged constants
Authored November 23, 2022
Make `jnp.copy` work with all shardings especially `PmapSharding`. This fixes the problem where a `jax.Array` with a `PmapSharding` round tripped through host and returned a `jax.Array` with a `SingleDeviceSharding`.
Authored December 7, 2022
Optimize `host_local_array_to_global_array` by caching the local to global conversion and flattening of axis resources. Also take a fast path for device_put which does not do `abstractify` and only canonicalize_dtype on the entire array once (instead of doing it for every shard).
Authored November 21, 2022
Mark arguments to ufuncs as positional-only.
Authored December 6, 2022
Remove opt-barrier fallbacks.
Authored November 17, 2022
[x64] more type safety for lax_numpy-related tests
Authored December 1, 2022
[x64] make chack_grads() more type-safe
Authored December 2, 2022
[TPU] Add cutoff for nearly diagonal matrices in QDWH-eigh algorithm.
Authored November 23, 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

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.4.1 Released December 13, 2022
0 Ξ”
jax-v0.4.1-rc Released December 13, 2022
592 Ξ”
jax-v0.4.0 Released December 12, 2022
0 Ξ”
jax-v0.4.0-rc1 Released December 12, 2022
2,877 Ξ”
jax-v0.4.0-rc Released December 8, 2022
14,191 Ξ”
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
7,610 Ξ”

Top Contributors in jax-v0.4.0-rc

jakevdp
LenaMartens
yashk2810
zhangqiaorjc
a-googler
hawkinsp
pschuh
froystig
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