Home » Oreps » Jax » All Releases » jax-v0.4.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
Releasesjax-v0.4.33
Basic Details Issues Addressed Top Contributors Directory Browser Release Notes
Jax jax-v0.4.33

Jax: jax-v0.4.33 Release

Release date:
September 16, 2024
Previous version:
jax-v0.4.33-rc (released September 16, 2024)
Magnitude:
0 Diff Delta
Contributors:
0 total committers
Data confidence:
Commits:
0 impacting commits

Top Contributors in jax-v0.4.33

Could not determine top contributors for this release.

Directory Browser for jax-v0.4.33

All files are compared to previous version, jax-v0.4.33-rc. Click here to browse diffs between other versions.

Loading File Browser...

Release Notes Published

This is a patch release on top of jax 0.4.32, that fixes two bugs found in that release.

A TPU-only data corruption bug was found in the version of libtpu pinned by JAX 0.4.32, which manifested only if multiple TPU slices were present in the same job, for example, if training on multiple v5e slices.

This release fixes that issue by pinning a fixed version of libtpu-nightly.

This release also fixes an inaccurate result for F64 tanh on CPU (#23590).

GitClear uses cookies to ensure you get the best experience on our website. Learn more