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

Jax: jaxlib-v0.1.33 Release

Release date:
April 16, 2025
Previous version:
jax-v0.6.0 (released April 22, 2025)
Magnitude:
451,688 Diff Delta
Contributors:
335 total committers
Data confidence:
Commits:
7,411 impacting commits

7411 Commits in this Release

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

Delete `xmap` and the `jax.experimental.maps` module. It's been 5 months since its deprecation (more than the standard 3 months deprecation period).
Authored July 24, 2024
Make trace dispatch purely a function of context rather than a function of both context and data. This lets us delete a lot of machinery for managing data-dependent tracing: levels, sublevels, post_process_call, new_base_main, custom_bind and so on.
Authored October 29, 2024
Upgrade JAX Advanced Autodiff 201
Authored December 15, 2023
[shard-map] add user tutorial
Authored January 8, 2024
Apply pyupgrade --py39-plus.
Authored July 21, 2023
[Pallas] Add experimental (private for now) API for manual fusion into Pallas kernels
Authored March 4, 2025
[Pallas] Add pallas distributed computation tutorial
Authored August 6, 2024
Remove legacy CPU custom call kernels that have been unused since v0.4.34.
Authored April 11, 2025
refactor jax.lax to not depend on jax.numpy
Authored September 1, 2024
Update flags to use the ABSL typed flag API.
Authored July 27, 2023
[shape_poly] Copy many of the jax2tf/shape_poly_test to live outside of jax2tf.
Authored November 19, 2023
Refactor pipeline emitter API.
Authored May 8, 2024
Add jax.experimental._mini_mpmd
Authored February 13, 2025
[Pallas TPU] Refactor ref indexers to transforms and support ref bitcast.
Authored September 16, 2024
Added ir.Value-based versions of some triton.compat APIs
Authored February 13, 2024
Now that sharding_in_types config flag is True, remove the config and all the conditionals
Authored February 19, 2025
[Pallas] Refactor indexing primitives to use NDIndexer abstraction
Authored January 2, 2024
[Pallas] Refactor pipelining docs.
Authored April 16, 2025
[shape_poly] Add limited support for equality explicit constraints.
Authored January 27, 2024
Some lax.linalg housekeeping.
Authored February 10, 2025
Add Mosaic to Jaxlib and expose bindings in `jax.experimental.mosaic`
Authored July 21, 2023
Start a new TPU interpret mode for Pallas.
Authored November 22, 2024
[shape_poly] Cleaning up naming of terms and factors.
Authored February 20, 2024
[shape_poly] Add a decision procedure for inequalities.
Authored January 20, 2024
Make changes to shard_map to prepare for setting `varying_axes_in_types` to True.
Authored April 8, 2025
[JAX] [XLA:Python] Migrate xla_extension and its type stubs into jaxlib.
Authored March 22, 2025
[sharding_in_types] Upstream changes from defaulting sharding_in_types config to True experiment. There aren't a lot of failures in TGP but we can atleast upstream these changes until we work on the failures.
Authored January 28, 2025
Pallas now uses MLIR Python builders to lower to Triton IR
Authored January 11, 2024
Add ensure_arraylike utility for lax.numpy implementations
Authored January 17, 2025
First pass at ufunc interfaces for several jax.numpy functions
Authored August 30, 2024
[host_callback] Remove obsolete tests.
Authored October 2, 2024
Upgrade JAX pytrees doc
Authored December 7, 2023
[export] Create the jax.export module APIs.
Authored June 10, 2024
Replace disable_backends with enable_backends on jax_multiplatform_test.
Authored September 27, 2024
Refactor gpusolver kernel definitions into separate build target.
Authored September 12, 2024
[Pallas TPU] Add support for passing in and returning semaphores
Authored September 19, 2024
Enable pivoted QR on CPU devices.
Authored December 4, 2024
Open source TPU-friendly ragged paged attention kernel.
Authored March 6, 2025
Move memories_test.py to JAX
Authored September 12, 2023
Upgrade How to think in JAX
Authored December 13, 2023

Browse Other Releases

Latest Pending Unreleased 😎
jaxlib-v0.1.40 Released March 4, 2020
0 Ξ”
jaxlib-v0.1.39 Released February 14, 2020
0 Ξ”
jaxlib-v0.1.38 Released January 29, 2020
0 Ξ”
jaxlib-v0.1.37 Released December 6, 2019
0 Ξ”
jaxlib-v0.1.36 Released November 22, 2019
0 Ξ”
jaxlib-v0.1.33 Released April 16, 2025
451,688 Ξ”
jax-v0.6.0 Released April 22, 2025
57,866 Ξ”
jax-v0.5.3 Released March 19, 2025
42,198 Ξ”
jax-v0.5.2 Released March 4, 2025
21 Ξ”
jax-v0.5.1 Released February 26, 2025
60,641 Ξ”

Top Contributors in jaxlib-v0.1.33

jakevdp
yashk2810
gnecula
hawkinsp
superbobry
apaszke
a-googler
dfm
mattjj
sharadmv

Directory Browser for jaxlib-v0.1.33

All files are compared to previous version, jax-v0.6.0. 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