Home » Oreps » Jax » All Releases » 0.3.20 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
Releases0.3.20
Basic Details Issues Addressed Top Contributors Directory Browser
Jax 0.3.20

Jax: 0.3.20 Release

Release date:
February 5, 2024
Previous version:
jaxlib-v0.4.25 (released February 24, 2024)
Magnitude:
132,070 Diff Delta
Contributors:
148 total committers
Data confidence:
Commits:
2,221 impacting commits

2221 Commits in this Release

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

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
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
Added ir.Value-based versions of some triton.compat APIs
Authored February 13, 2024
[Pallas] Refactor indexing primitives to use NDIndexer abstraction
Authored January 2, 2024
[shape_poly] Add limited support for equality explicit constraints.
Authored January 27, 2024
Add Mosaic to Jaxlib and expose bindings in `jax.experimental.mosaic`
Authored July 21, 2023
[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
[shape_poly] Improve compile-time shape checking.
Authored May 27, 2023
Pallas now uses MLIR Python builders to lower to Triton IR
Authored January 11, 2024
Upgrade JAX pytrees doc
Authored December 7, 2023
Move memories_test.py to JAX
Authored September 12, 2023
Upgrade How to think in JAX
Authored December 13, 2023
[shape_poly] Performance improvements for symbolic dimension manipulations (step 3)
Authored February 15, 2024
[shape_poly] Add more tests for reasoning about inequalities.
Authored January 24, 2024
Upgrade JAX Installation doc
Authored December 4, 2023
Migrate another subset of internal modules to use state objects
Authored October 11, 2023
[Mosaic] apply_vector_layout C++ rewrite (5) vector load and store
Authored September 20, 2023
[MOSAIC] apply_vector_layout C++ rewrite (3) applyLayoutOp and relayout
Authored September 7, 2023
[callback] Disable stream_executor tests.
Authored August 22, 2023
[shape_poly] Improve error messages for shape assertions
Authored July 21, 2023
[key reuse] remove alternate simple implementation
Authored February 15, 2024
Remove code that disabled tests on "stream_executor" backends.
Authored September 14, 2023
[Pallas/Mosaic] Add support for nested `ref.at`
Authored January 3, 2024
Use MLIR generated convenience functions athing(...) instead of writing AThingOp(...).result.
Authored November 17, 2023
Switch jaxlib to use nanobind instead of pybind11.
Authored August 24, 2023
[Mosaic] Python bindings for `VectorLayout`, `VRegDataBounds`, `assemble`, `disassemble`, `relayout` and `apply_layout_op`
Authored November 21, 2023
MAINT Clean up leftover `Array = Any` aliases in jax/_src/**.py
Authored September 25, 2023
[PJRT C API] Set up jax xla cuda package.
Authored September 13, 2023
[export] Move jax_export and shape_poly out of jax2tf.
Authored September 6, 2023
output res forwarding optimization for shard_map and jit
Authored October 12, 2023
[shape_poly] Fixes for the lexicographic ordering of monomials.
Authored January 5, 2024
[export] Fix the call_exported in presence of shardings.
Authored October 6, 2023
[Mosaic] Remove Python implementation of apply_vector_layout and infer_memref_layout.
Authored January 10, 2024
[Memories] Add Memories support to jax.jit and jax.device_put!
Authored August 4, 2023
Upgrade most .py sources to 3.9
Authored December 8, 2023
[export] Add support for serialization and deserialization of Exported
Authored December 12, 2023

Browse Other Releases

Latest Pending Unreleased 😎
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 Ξ”
jaxlib-v0.4.22 Released December 13, 2023
7,566 Ξ”

Top Contributors in 0.3.20

gnecula
jakevdp
hawkinsp
yashk2810
mattjj
superbobry
sharadmv
tlongeri
8bitmp3
a-googler

Directory Browser for 0.3.20

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