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

Jax: jaxlib-v0.3.25 Release

Release date:
November 15, 2022
Previous version:
jaxlib-v0.3.24 (released November 4, 2022)
Magnitude:
6,770 Diff Delta
Contributors:
27 total committers
Data confidence:
Commits:
92 impacting commits

92 Commits in this Release

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

add a basic prototype of piles, behind jax_dynamic_shapes
Authored October 17, 2022
Add support for Hessenberg and tridiagonal matrix reductions on CPU
Authored November 9, 2022
Add jax.Array migration doc to OSS
Authored November 11, 2022
Delete `jax_experimental_name_stack` flag
Authored November 10, 2022
Add a number of missing function cross-references in the docs
Authored August 24, 2022
Add a GPU implementation of symmetric (Hermitian) tridiagonal reduction
Authored November 10, 2022
Rename test flag --num_generated_cases to --jax_num_generated_cases
Authored November 10, 2022
Make `MeshPspecSharding` an alias for `NamedSharding` (it was the other way around before this CL)
Authored November 14, 2022
Add UnloadedMeshExecutable to represent a MeshExecutable that is not loaded
Authored November 7, 2022
Add docstrings for `Sharding` classes. 0
Authored November 14, 2022
[sparse] add bcoo_gather & support for sparse indexing
Authored November 11, 2022
[linalg] Add jax.scipy.special.bessel_jn (Bessel function of the first kind)
Authored November 9, 2022
[jax2tf] Fixed jax2tf Limitations
Authored November 7, 2022
[sparse] support strides in bcoo_slice
Authored November 9, 2022
[sparse] streamline sparse rules for standard primitives
Authored November 11, 2022
Change pickling for jax.sharding to not serialize device ids
Authored November 11, 2022
tighten up vmap w/ piles: require pile_axis in_axes/out_axes
Authored November 7, 2022
sparse] BCSR batching rule
Authored November 10, 2022
Add support for the hermitian option on jnp.linalg.pinv
Authored November 7, 2022
Avoid forming identity matrix in SVD JVP
Authored November 7, 2022
[sparse] add bcoo_squeeze function
Authored November 9, 2022
Set default layout for Python callback
Authored November 10, 2022
Don't donate arguments in jit/pmap/pjit when debug_nans=True
Authored November 7, 2022
Introduce XlaLowering::stablehlo() and use it in associated APIs
Authored November 9, 2022
Add docstrings for jax.Array APIs `make_array_from_callback` and `make_array_from_single_device_arrays`
Authored November 11, 2022
Checkify: fix nan_checks+PRNGKeys - a PRNGKey is never NaN!
Authored November 9, 2022
Convert string axis name into tuple of strings in Mesh constructor
Authored November 11, 2022
[sparse] bcoo_dynamic_slice: remove unnecessary padding from output
Authored November 9, 2022
If the input to `host_local_array_to_global_array` is not fully addressable (i.e
Authored November 6, 2022
Add a section to mention the `unsafe_rbg` approach for reducing the size of the generated TF graph
Authored November 4, 2022
[sparse] add support for BCOO.astype method
Authored November 8, 2022
[sparse] fix shape bug in bcoo_transpose
Authored November 9, 2022
Improve the error message when `@pjit` (with no {in_axis|out_axis}_resources is used without jax.Array enabled
Authored November 10, 2022
Add tf.get_current_name_scope() as prefix of name_stack during tf execution
Authored November 14, 2022
Disable more tests under tsan/asan
Authored November 14, 2022
[sparse] Add BCSR from_scipy_sparse
Authored November 10, 2022
Remove `device_indices` method which is redundant because of the existence of `devices_indices_map` and is slower because pinging a cache for every device is not free
Authored November 6, 2022
Make Sharding instances picklable
Authored November 5, 2022
More documentation about GPU tests
Authored November 1, 2022
Remove static_argnums from AOT invocation
Authored November 7, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.4 Released February 16, 2023
5,774 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
12,125 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
25,848 Ξ”
jaxlib-v0.4.1 Released December 13, 2022
501 Ξ”
jaxlib-v0.4.0 Released December 12, 2022
16,767 Ξ”
jaxlib-v0.3.25 Released November 15, 2022
6,770 Ξ”
jaxlib-v0.3.24 Released November 4, 2022
20,656 Ξ”
jaxlib-v0.3.22 Released October 11, 2022
16 Ξ”
jaxlib-v0.3.20 Released September 28, 2022
49,762 Ξ”
jaxlib-v0.3.15 Released July 22, 2022
290 Ξ”

Top Contributors in jaxlib-v0.3.25

hawkinsp
yashk2810
mattjj
sharadmv
jakevdp
pschuh
tlu7
LenaMartens
gnecula
burmako

Directory Browser for jaxlib-v0.3.25

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