Home » Oreps » Jax » All Releases » pending_release 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
ReleasesLatest Pending
Basic Details Issues Addressed Top Contributors Directory Browser
Jax Latest Pending

Jax: Latest Pending Release

Release date:
Pending release in a future version
Previous version:
0.3.25 (released November 15, 2022)
Magnitude:
464,364 Diff Delta
Contributors:
372 total committers
Data confidence:
---
Commits:
4,528 impacting commits

4528 Commits in this Release

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

Stackless yashful
Authored October 2, 2024
more documentation updates; include links to code instead of inlined code
Authored March 12, 2024
[host_callback] Remove most of the jax.experimental.host_callback module
Authored October 5, 2024
Enable building jaxlib w/ Mosaic
Authored July 20, 2023
Add new scripts and configs for building JAX artifacts
Authored September 16, 2024
Changed lower_jaxpr_to_triton_ir to use ir.Value instead of triton.compat APIs
Authored February 9, 2024
[Pallas/Mosaic] Add support for nested `ref.at`
Authored December 29, 2023
Regenerate python stubs
Authored May 20, 2024
[Mosaic] apply_vector_layout C++ rewrite: Fix check for defining layout in disassemble
Authored September 27, 2023
[Mosaic] apply_vector_layout C++ rewrite (8): tpu.gather, tpu.iota, tpu.trace
Authored August 21, 2023
[Mosaic] apply_vector_layout C++ rewrite (10): vector.extract_strided_slice
Authored September 6, 2023
[MOSAIC] apply_vector_layout C++ rewrite (7): tpu load and store
Authored August 9, 2023
[Mosaic] apply_vector_layout C++ rewrite (9): tpu.repeat
Authored September 5, 2023
Refactor pipeline emitter API.
Authored April 23, 2024
[MOSAIC] apply_vector_layout rewrite (6): arith.constant
Authored August 9, 2023
Add new scripts and configs for running JAX tests
Authored September 16, 2024
[Mosaic] apply_vector_layout C++ rewrite (19): vector.transpose
Authored October 4, 2023
Now that sharding_in_types config flag is True, remove the config and all the conditionals
Authored February 19, 2025
init
Authored May 14, 2024
init
Authored May 20, 2024
[shard-map] add rewrite for efficient transposition
Authored May 10, 2023
Some lax.linalg housekeeping.
Authored February 10, 2025
[export] Add support for serialization and deserialization of Exported
Authored November 30, 2023
Start a new TPU interpret mode for Pallas.
Authored November 22, 2024
[Mosaic] apply_vector_layout C++ rewrite: Enable apply_vector_layout_test.py
Authored October 19, 2023
[JAX] Improve handling of metadata in compilation cache.
Authored April 17, 2023
migrate more internal dependencies from `jax.core` to `jax._src.core`
Authored February 11, 2023
[Pallas TPU] Add matrix multiplication tutorial
Authored February 28, 2024
Avoid imports from the public jax.* namespace in more places internally.
Authored April 4, 2023
[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
[Mosaic] apply_vector_layout C++ rewrite (16): vector.multi_reduction
Authored September 26, 2023
[2/n] Embed the tf.Graph into the stablehlo.custom_call.
Authored April 13, 2023
Scaled matmul for mxfp8
Authored February 5, 2025
[Memories] Add Memories support to jax.jit and jax.device_put!
Authored July 19, 2023
Use mpmath functions as a reference for evaluating the correctness of jax complex functions
Authored March 21, 2024
DOC: add manual documentation to jax.scipy.special functions
Authored April 22, 2024
Simplify and consolidate dot algorithm control in lax.
Authored October 2, 2024
Use hermetic Python in JAX, see "Managing hermetic Python" in developer.md for details
Authored March 28, 2024
Bump minimum_jaxlib_version to 0.4.14. `xla_extension_version` is 174 and `mlir_api_version` is 54
Authored July 28, 2023
[pallas] Move ops_test.py from jax_triton to jax/pallas
Authored August 9, 2024

Browse Other Releases

Latest Pending Unreleased 😎
0.3.25 Released November 15, 2022
7,635 Ξ”
0.3.24 Released November 4, 2022
22,988 Ξ”
0.3.23 Released October 12, 2022
305 Ξ”
0.3.22 Released October 11, 2022
11,621 Ξ”
0.3.21 Released September 30, 2022
1,287 Ξ”
0.3.20 Released September 28, 2022
0 Ξ”
jaxlib-v0.4.32 Released September 11, 2024
0 Ξ”
jaxlib-v0.4.31 Released July 29, 2024
41,440 Ξ”
jaxlib-v0.4.30 Released June 18, 2024
10,316 Ξ”
jaxlib-v0.4.29 Released June 10, 2024
27,453 Ξ”

Top Contributors in Latest Pending

gnecula
yashk2810
a-googler
jakevdp
tlongeri
dougalm
hawkinsp
apaszke
superbobry
dfm

Directory Browser for Latest Pending

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