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:
20,648 Diff Delta
Contributors:
35 total committers
Data confidence:
Commits:
200 impacting commits

200 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
[jax2tf] An alternative support for shape polymorphism for native serialization
Authored November 28, 2022
Add CUDNN custom call for LSTM. 0
Authored November 23, 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
DOC: add example of pure_callback with custom_jvp
Authored December 9, 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
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. 0
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
Add JAX monitoring library that instruments code via events
Authored November 15, 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
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 local imports of array.py. 0
Authored November 30, 2022
Remove unused potrf kernels
Authored November 17, 2022
Move `PartitionSpec` and `Mesh` out of experimental and into the `sharding` namespace
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
[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`
Authored December 7, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.6 Released March 9, 2023
21,059 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
6,670 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
14,421 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
31,232 Ξ”
jaxlib-v0.4.1 Released December 13, 2022
592 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
20,648 Ξ”
jaxlib-v0.3.25 Released November 15, 2022
8,314 Ξ”
jaxlib-v0.3.24 Released November 4, 2022
22,228 Ξ”
jaxlib-v0.3.22 Released October 11, 2022
22 Ξ”
jaxlib-v0.3.20 Released September 28, 2022
58,445 Ξ”

Top Contributors in jaxlib-v0.4.0

jakevdp
yashk2810
LenaMartens
gnecula
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 about our cookie policy