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

Jax: jaxlib-v0.4.0 Release

Release date:
December 12, 2022
Previous version:
jaxlib-v0.3.25 (released November 15, 2022)
Magnitude:
17,050 Diff Delta
Contributors:
35 total committers
Data confidence:
Commits:
197 impacting commits

197 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
[jax2tf] An alternative support for shape polymorphism for native serialization.
Authored November 28, 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
DOC: add example of pure_callback with custom_jvp
Authored December 9, 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
Improves handling of opaque types for dynamic shapes
Authored December 4, 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
in jax.Array notebook, polish beginning and tweak title and some wording
Authored December 11, 2022
[x64] update api_test for type safety
Authored November 29, 2022
Make JAX tests pass under NumPy 1.24.0rc2.
Authored December 8, 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

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.4 Released February 16, 2023
6,660 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
8,565 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
26,178 Ξ”
jaxlib-v0.4.1 Released December 13, 2022
592 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
17,050 Ξ”
jaxlib-v0.3.25 Released November 15, 2022
7,636 Ξ”
jaxlib-v0.3.24 Released November 4, 2022
23,851 Ξ”
jaxlib-v0.3.22 Released October 11, 2022
12,900 Ξ”
jaxlib-v0.3.20 Released September 28, 2022
56,163 Ξ”
jaxlib-v0.3.15 Released July 22, 2022
16,976 Ξ”

Top Contributors in jaxlib-v0.4.0

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

Directory Browser for jaxlib-v0.4.0

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