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:
257,851 Diff Delta
Contributors:
250 total committers
Data confidence:
---
Commits:
2,353 impacting commits

2353 Commits in this Release

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

DOC: replace old tutorials with new content
Authored April 18, 2024
more documentation updates; include links to code instead of inlined code
Authored March 12, 2024
Enable building jaxlib w/ Mosaic
Authored July 20, 2023
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
[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
[MOSAIC] apply_vector_layout rewrite (6): arith.constant
Authored August 9, 2023
[Mosaic] apply_vector_layout C++ rewrite (19): vector.transpose
Authored October 4, 2023
jax.scipy.stats: manually document functions to avoid scipy import
Authored April 19, 2024
[shard-map] add rewrite for efficient transposition
Authored May 10, 2023
Delete docs/jep/9263-typed-keys.md
Authored April 2, 2024
[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
Avoid imports from the public jax.* namespace in more places internally.
Authored April 4, 2023
[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
Bump minimum_jaxlib_version to 0.4.14. `xla_extension_version` is 174 and `mlir_api_version` is 54
Authored July 28, 2023
Use hermetic Python in JAX, see "Managing hermetic Python" in developer.md for details
Authored March 28, 2024
[jax2tf] First draft of supporting multi-platform serialization
Authored December 27, 2022
Add experimental static key reuse checking
Authored December 8, 2023
[Mosaic] apply_vector_layout C++ rewrite (15): vector.shape_cast
Authored September 25, 2023
[export] Add support for exporting functions with effects
Authored August 25, 2023
Inlined some Triton-specific abstractions
Authored February 1, 2024
Add `Layout` support to `jax.jit`.
Authored April 6, 2024
Use MLIR generated convenience functions athing(...) instead of writing AThingOp(...).result.
Authored November 15, 2023
[Memories][PJRT:C] Add PJRT_Memory to the pjrt c api.
Authored August 1, 2023
[Mosaic] apply_vector_layout C++ rewrite (14): tpu.matmul and vector.contract
Authored September 25, 2023
Add experimental static key reuse checking
Authored November 2, 2023
Added scipy.spatial.transform Rotation and Slerp classes
Authored June 3, 2023
Upgrade JAX Installation doc
Authored November 28, 2023
[jax2tf] Refactor jax2tf to move more pure-JAX stuff to jax_export
Authored April 4, 2023

Browse Other Releases

Latest Pending Unreleased 😎
0.3.25 Released November 15, 2022
7,636 Ξ”
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 February 5, 2024
132,070 Ξ”
jaxlib-v0.4.26 Released April 3, 2024
16,870 Ξ”
jaxlib-v0.4.25 Released February 24, 2024
17,299 Ξ”
jaxlib-v0.4.24 Released February 5, 2024
29,867 Ξ”
jaxlib-v0.4.23 Released December 14, 2023
930 Ξ”

Top Contributors in Latest Pending

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

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