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

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:

475,787 Diff Delta

Contributors:

379 total committers

Data confidence:

---

Commits:

4,832

4832 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

[Pallas] Add pallas distributed computation tutorial

Authored August 6, 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

refactor jax.lax to not depend on jax.numpy

Authored September 1, 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

Refactor pipeline emitter API.

Authored May 8, 2024

[Mosaic] apply_vector_layout C++ rewrite (19): vector.transpose

Authored October 4, 2023

init

Authored May 14, 2024

init

Authored May 20, 2024

[shard-map] add rewrite for efficient transposition

Authored May 10, 2023

[export] Add support for serialization and deserialization of Exported

Authored November 30, 2023

[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

[JAX] [XLA:Python] Migrate xla_extension and its type stubs into jaxlib.

Authored March 22, 2025

[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

First pass at ufunc interfaces for several jax.numpy functions

Authored August 30, 2024

[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

[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

a-googler

jakevdp

yashk2810

tlongeri

hawkinsp

dougalm

superbobry

apaszke

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.