Home » Oreps » Jax » All Releases » 0.3.25 Release
google/jax
Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more
Watch
0
Commit Activity
Directory Browser
Velocity History
Releases0.3.25
Basic Details Issues Addressed Top Contributors Directory Browser
Jax 0.3.25

Jax: 0.3.25 Release

Release date:
November 15, 2022
Previous version:
0.3.24 (released November 4, 2022)
Magnitude:
6,881 Diff Delta
Contributors:
27 total committers
Data confidence:
Commits:
90 impacting commits

90 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
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
[jax2tf] Fixed jax2tf Limitations
Authored November 7, 2022
[sparse] add bcoo_gather & support for sparse indexing
Authored November 11, 2022
Add docstrings for `Sharding` classes. Right now I am only documenting `Sharding`, `XLACompatibleSharding`, `MeshPspecSharding` and `SingleDeviceSharding`.
Authored November 14, 2022
tighten up vmap w/ piles: require pile_axis in_axes/out_axes
Authored November 7, 2022
[linalg] Add jax.scipy.special.bessel_jn (Bessel function of the first kind).
Authored November 9, 2022
Introduce XlaLowering::stablehlo() and use it in associated APIs
Authored November 9, 2022
Add support for the hermitian option on jnp.linalg.pinv.
Authored November 7, 2022
Change pickling for jax.sharding to not serialize device ids.
Authored November 11, 2022
Set default layout for Python callback
Authored November 10, 2022
[sparse] support strides in bcoo_slice
Authored November 9, 2022
sparse] BCSR batching rule.
Authored November 10, 2022
[sparse] streamline sparse rules for standard primitives
Authored November 11, 2022
Avoid forming identity matrix in SVD JVP.
Authored November 7, 2022
[sparse] add bcoo_squeeze function
Authored November 9, 2022
Disable asan/tsan for a number of slow tests.
Authored November 14, 2022
Don't donate arguments in jit/pmap/pjit when debug_nans=True.
Authored November 7, 2022
Make Sharding instances picklable.
Authored November 5, 2022
[ROCm]: Enable/update multiprocess gpu tests for ROCm
Authored October 20, 2022
Convert string axis name into tuple of strings in Mesh constructor
Authored November 11, 2022
Add tf.get_current_name_scope() as prefix of name_stack during tf execution.
Authored November 14, 2022
[sparse] fix shape bug in bcoo_transpose
Authored November 9, 2022
Add a section to mention the `unsafe_rbg` approach for reducing the size of the generated TF graph.
Authored November 4, 2022
[sparse] bcoo_dynamic_slice: remove unnecessary padding from output
Authored November 9, 2022
Disable more tests under tsan/asan.
Authored November 14, 2022
Checkify: fix nan_checks+PRNGKeys - a PRNGKey is never NaN!
Authored November 9, 2022
If the input to `host_local_array_to_global_array` is not fully addressable (i.e. not host local), return it as is.
Authored November 6, 2022
Add docstrings for jax.Array APIs `make_array_from_callback` and `make_array_from_single_device_arrays`.
Authored November 11, 2022
PRNGKeyArray is now a virtual subclass of ndarray
Authored November 11, 2022
Let is_user_frame ignore frames from stdlib.
Authored November 8, 2022
[sparse] Add BCSR from_scipy_sparse.
Authored November 10, 2022
[sparse] add support for BCOO.astype method
Authored November 8, 2022
Improve the error message when `@pjit` (with no {in_axis|out_axis}_resources is used without jax.Array enabled.
Authored November 10, 2022
More documentation about GPU tests.
Authored November 1, 2022

Browse Other Releases

Latest Pending Unreleased 😎
0.3.25 Released November 15, 2022
6,881 Ξ”
0.3.24 Released November 4, 2022
23,084 Ξ”
0.3.23 Released October 12, 2022
305 Ξ”
0.3.22 Released October 11, 2022
11,644 Ξ”
0.3.21 Released September 30, 2022
1,561 Ξ”
0.3.20 Released March 9, 2023
101,633 Ξ”
jaxlib-v0.4.6 Released March 9, 2023
17,010 Ξ”
jaxlib-v0.4.4 Released February 16, 2023
6,724 Ξ”
jaxlib-v0.4.3 Released February 8, 2023
7,645 Ξ”
jaxlib-v0.4.2 Released January 25, 2023
24,737 Ξ”

Top Contributors in 0.3.25

hawkinsp
mattjj
yashk2810
sharadmv
jake-vanderplas-1c1d
pschuh
tlu7
gnecula
LenaMartens
burmako

Directory Browser for 0.3.25

All files are compared to previous version, 0.3.24. Click here to browse diffs between other versions.

Loading File Browser...
GitClear uses cookies to ensure you get the best experience on our website. Learn more