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

Jax: jaxlib-v0.4.13 Release

Release date:
June 22, 2023
Previous version:
jaxlib-v0.4.12 (released June 8, 2023)
Magnitude:
7,868 Diff Delta
Contributors:
25 total committers
Data confidence:
Commits:
119 impacting commits

119 Commits in this Release

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

[shape_poly] Improve compile-time shape checking.
Authored May 27, 2023
[shape_poly] DynamicReduceWindow: add shape poly support with native serialization.
Authored June 17, 2023
Add support for dynamic shapes to jax.fft.
Authored June 17, 2023
Fix some test failures under H100.
Authored June 9, 2023
Allow None to be passed to in_shardings and out_shardings. The default is still UNSPECIFIED to handle edge cases around the old semantics where None is treated as fully replicated.
Authored June 15, 2023
[jax2tf] Add native_lowering_disabled_checks parameter to jax2tf.convert.
Authored June 10, 2023
[shape_poly] DynamicRngBitGenerator: add shape poly support with native serialization.
Authored June 19, 2023
Deduplicate references closed over across branches of a lax.cond.
Authored June 16, 2023
[shape_poly] linalg.eig: shape polymorphism with native serialization on CPU
Authored June 17, 2023
[jax2tf] Refactor the backward compatibility tests.
Authored June 19, 2023
[shape_poly] Refactor and simplify tests.
Authored June 16, 2023
Raise a RuntimError if plugin initialization fails, rather than logging at info priority.
Authored June 14, 2023
jax errors: improve leading info in error docs
Authored June 14, 2023
[shape_poly] linalg.svd: shape polymorphism for native serialization on CPU
Authored June 22, 2023
Fix symbol collision when merging MLIR modules.
Authored June 20, 2023
Test and implement broadcast_in_dim for all permutations of ragged axes.
Authored June 4, 2023
[jax2tf] Refactor the backwards compatibility tests.
Authored June 22, 2023
Added argument jax.linearize(..., has_aux=...)
Authored May 18, 2023
Add a link to the Apple Metal plugin to the JAX README.
Authored June 21, 2023
Change the representation of RaggedAxis to allow multiple ragged axes under the same stacked axis.
Authored June 5, 2023
Add a warning if a non-allowlisted plugin is used.
Authored June 20, 2023
Fix corner cases in JAX SVD: a) Clamp negative singular values to zero. b) Return all NaN for matrices with non-finite values.
Authored June 13, 2023
Fix problematic rotation tests
Authored June 20, 2023
[XLA:Python] Fix __cuda_array_interface__.
Authored June 20, 2023
Fix jax2tf graph_serialization irfft length issue.
Authored June 11, 2023
Update Windows CI.
Authored June 12, 2023
Add support for Github Actions build of jaxlib on Winx64
Authored March 15, 2023
Test and implement squeeze under ragged batching.
Authored June 6, 2023
random.key: error for non-scalar seeds.
Authored June 20, 2023
Increase absolute tolerances in eigh test.
Authored June 12, 2023
Better pprint rule for check_p primitive.
Authored June 12, 2023
Fix mypy error.
Authored June 8, 2023
Fix test failure in LaxBackedNumpyTest.testFrexp4 on Windows.
Authored June 14, 2023
Raise a good error message when a ShapeDtypeStruct is closed over as a const which is not a valid arg during execution.
Authored June 14, 2023
Windows CI improvements:
Authored June 13, 2023
Fix xla_bridge_test.py test failures.
Authored June 16, 2023
Add live-analysis memory optimization to more jaxpr interpreters.
Authored June 16, 2023
[shape_poly] Change lowering for shape polymorphism to simplify shape refinement
Authored June 15, 2023
Return PositionalSharding instead of GSPMDSharding in custom_partitioning when mesh is not defined
Authored June 12, 2023
Jax 0.4.13 release.
Authored June 22, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.18 Released October 6, 2023
3,149 Ξ”
jaxlib-v0.4.17 Released October 3, 2023
8,498 Ξ”
jaxlib-v0.4.16 Released September 18, 2023
9,916 Ξ”
jaxlib-v0.4.15 Released August 30, 2023
14,550 Ξ”
jaxlib-v0.4.14 Released July 27, 2023
19,891 Ξ”
jaxlib-v0.4.13 Released June 22, 2023
7,868 Ξ”
jaxlib-v0.4.12 Released June 8, 2023
4,680 Ξ”
jaxlib-v0.4.11 Released May 31, 2023
6,799 Ξ”
jaxlib-v0.4.10 Released May 11, 2023
1,233 Ξ”
jaxlib-v0.4.9 Released May 9, 2023
25,643 Ξ”

Top Contributors in jaxlib-v0.4.13

gnecula
hawkinsp
yashk2810
axch
jakevdp
sharadmv
LenaMartens
patrick-kidger
a-googler
chrisflesher

Directory Browser for jaxlib-v0.4.13

We haven't yet finished calculating and confirming the files and directories changed in this release. Please check back soon.

Release Notes Published

  • Changes

    • Added Windows CPU-only wheels to the jaxlib Pypi release.

  • Bug fixes

    • __cuda_array_interface__ was broken in previous jaxlib versions and is now fixed ({jax-issue}16440).
    • Concurrent CUDA kernel tracing is now enabled by default on NVIDIA GPUs.
GitClear uses cookies to ensure you get the best experience on our website. Learn more