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

Jax: jax-v0.3.18 Release

Release date:
September 26, 2022
Previous version:
jax-v0.3.18-rc (released September 26, 2022)
Magnitude:
785 Diff Delta
Contributors:
4 total committers
Data confidence:
Commits:
6 impacting commits

Top Contributors in jax-v0.3.18

jakevdp
LenaMartens
mattjj
inoryy

Directory Browser for jax-v0.3.18

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

Release Notes Published

  • GitHub commits.
  • Changes
    • Ahead-of-time lowering and compilation functionality (tracked in {jax-issue}#7733) is stable and public. See the overview and the API docs for {mod}jax.stages.
    • Introduced {class}jax.Array, intended to be used for both isinstance checks and type annotations for array types in JAX. Notice that this included some subtle changes to how isinstance works for {class}jax.numpy.ndarray for jax-internal objects, as {class}jax.numpy.ndarray is now a simple alias of {class}jax.Array.
  • Breaking changes
    • jax._src is no longer imported into the from the public jax namespace. This may break users that were using JAX internals.
    • jax.soft_pmap has been deleted. Please use pjit or xmap instead. jax.soft_pmap is undocumented. If it were documented, a deprecation period would have been provided.
GitClear uses cookies to ensure you get the best experience on our website. Learn more