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
Pull Requests
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:
6,666 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
Make JAX-Triton calls serializable
Authored June 22, 2023
Add support for dynamic shapes to jax.fft
Authored June 17, 2023
[shape_poly] DynamicReduceWindow: add shape poly support with native serialization
Authored June 17, 2023
Allow None to be passed to in_shardings and out_shardings. 0
Authored June 15, 2023
Fix some test failures under H100
Authored June 9, 2023
[jax2tf] Add native_lowering_disabled_checks parameter to jax2tf.convert
Authored June 10, 2023
Deduplicate references closed over across branches of a lax.cond
Authored June 16, 2023
[shape_poly] DynamicRngBitGenerator: add shape poly support with native serialization
Authored June 19, 2023
[shape_poly] linalg.eig: shape polymorphism with native serialization on CPU
Authored June 17, 2023
jax errors: improve leading info in error docs
Authored June 14, 2023
Raise a RuntimError if plugin initialization fails, rather than logging at info priority
Authored June 14, 2023
[jax2tf] Refactor the backward compatibility tests
Authored June 19, 2023
[shape_poly] Refactor and simplify tests
Authored June 16, 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
Change the representation of RaggedAxis to allow multiple ragged axes under the same stacked axis
Authored June 5, 2023
Added argument jax.linearize(..., has_aux=...)
Authored May 18, 2023
Fix corner cases in JAX SVD: a) Clamp negative singular values to zero
Authored June 13, 2023
Add a link to the Apple Metal plugin to the JAX README
Authored June 21, 2023
Update Windows CI
Authored June 12, 2023
[jax2tf] Refactor the backwards compatibility tests
Authored June 22, 2023
Add support for Github Actions build of jaxlib on Winx64
Authored March 15, 2023
Fix problematic rotation tests
Authored June 20, 2023
[XLA:Python] Fix __cuda_array_interface__
Authored June 20, 2023
Test and implement squeeze under ragged batching
Authored June 6, 2023
random.key: error for non-scalar seeds
Authored June 20, 2023
Add a warning if a non-allowlisted plugin is used
Authored June 20, 2023
Fix mypy error
Authored June 8, 2023
Windows CI improvements:
Authored June 13, 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
Fix jax2tf graph_serialization irfft length issue
Authored June 11, 2023
Increase absolute tolerances in eigh test
Authored June 12, 2023
Return PositionalSharding instead of GSPMDSharding in custom_partitioning when mesh is not defined
Authored June 12, 2023
Remove the monkey patch in jax2tf by moving the function to mlir.py
Authored June 10, 2023
Add a CI presubmit that renders the documentation
Authored June 16, 2023
Use cloudpickle for serializing host callbacks
Authored June 9, 2023
Fix test failure in LaxBackedNumpyTest.testFrexp4 on Windows
Authored June 14, 2023
Fix xla_bridge_test.py test failures
Authored June 16, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.4.18 Released October 6, 2023
4,220 Ξ”
jaxlib-v0.4.17 Released October 3, 2023
9,022 Ξ”
jaxlib-v0.4.16 Released September 18, 2023
9,933 Ξ”
jaxlib-v0.4.15 Released August 30, 2023
15,453 Ξ”
jaxlib-v0.4.14 Released July 27, 2023
17,116 Ξ”
jaxlib-v0.4.13 Released June 22, 2023
6,666 Ξ”
jaxlib-v0.4.12 Released June 8, 2023
3,820 Ξ”
jaxlib-v0.4.11 Released June 3, 2023
6,363 Ξ”
jaxlib-v0.4.10 Released June 3, 2023
1,057 Ξ”
jaxlib-v0.4.9 Released June 3, 2023
23,085 Ξ”

Top Contributors in jaxlib-v0.4.13

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

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 about our cookie policy