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

Jax: jax-v0.3.22 Release

Release date:
October 11, 2022
Previous version:
jax-v0.3.22-rc (released October 11, 2022)
Magnitude:
0 Diff Delta
Contributors:
0 total committers
Data confidence:
Commits:
0 impacting commits

Top Contributors in jax-v0.3.22

Could not determine top contributors for this release.

Directory Browser for jax-v0.3.22

We haven't yet finished calculating and confirming the files and directories changed in this release. Please check back soon.

Release Notes Published

  • Changes
    • Add JAX_PLATFORMS=tpu,cpu as default setting in TPU initialization, so JAX will raise an error if TPU cannot be initialized instead of falling back to CPU. Set JAX_PLATFORMS='' to override this behavior and automatically choose an available backend (the original default), or set JAX_PLATFORMS=cpu to always use CPU regardless of if the TPU is available.
  • Deprecations
    • Several test utilities deprecated in JAX v0.3.8 are now removed from {mod}jax.test_util.
GitClear uses cookies to ensure you get the best experience on our website. Learn more