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

Jax: jax-v0.3.6 Release

Release date:
April 13, 2022
Previous version:
jax-v0.3.5 (released April 7, 2022)
Magnitude:
5,775 Diff Delta
Contributors:
9 total committers
Data confidence:
Commits:
26 impacting commits

26 Commits in this Release

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

prototyping dynamic shapes
Authored March 31, 2022
remove experimental/djax
Authored April 8, 2022
Adds simple effect types to jaxprs
Authored February 28, 2022
[MHLO] Add direct MHLO lowerings for sparse primitives.
Authored April 8, 2022
Remove invertible_ad since it's not in use.
Authored April 11, 2022
[sparse] fix autodiff bug in spdot_general
Authored April 8, 2022
dispatch.py: type annotations, other minor tweaks
Authored April 9, 2022
[MHLO] Add a direct MHLO lowering for pjit_p, which lacked one.
Authored April 8, 2022
Revert: [MHLO] Add a direct MHLO lowering for pjit_p, which lacked one.
Authored April 8, 2022
[MHLO] Add a direct MHLO lowering for pjit_p, which lacked one.
Authored April 12, 2022
[JAX] Fix batch logic for approx_min/max_k
Authored April 8, 2022
Simplify `jnp.trace` implementation
Authored April 8, 2022
[sparse] Use sorted indices instead of sorted rows only.
Authored April 9, 2022
Simplify `jnp.diagonal` implementation
Authored April 8, 2022
Enable tests related to the Gamma distribution for non-default PRNG implementations only when jax_enable_custom_prng is enabled, for consistency with other tests.
Authored April 8, 2022
[MHLO] Switch sharded_jit dispatch path to use MHLO lowering.
Authored April 11, 2022
[JAX] Adds the approx_top_k_p bridge.
Authored April 12, 2022
Update version numbers after jax/jaxlib release.
Authored April 7, 2022
Roll forward manylinux2014 builds after fixes.
Authored April 9, 2022
Revert: Pin a newer @platforms in the Bazel workspace to fix Mac ARM builds.
Authored April 8, 2022
Fix the pytype error. PyType is looking for a __init__ method. This does not change the behavior of the class.
Authored April 12, 2022
Make jaxlib builds manylinux2014 compliant.
Authored April 8, 2022
Make jaxlib builds manylinux2014 compliant.
Authored April 9, 2022
Upgrade the bazel version to 5.1.1
Authored April 13, 2022
Roll forward again after the fix in the auto sharding pass.
Authored April 8, 2022
Bump the libtpu version to prepare for JAX release
Authored April 12, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.10 Released May 4, 2022
1,645 Ξ”
jax-v0.3.9 Released May 3, 2022
878 Ξ”
jax-v0.3.8 Released April 30, 2022
11,149 Ξ”
jax-v0.3.7 Released April 15, 2022
4,073 Ξ”
jax-v0.3.6 Released April 13, 2022
5,775 Ξ”
jax-v0.3.5 Released April 7, 2022
9,132 Ξ”
jax-v0.3.4 Released March 18, 2022
331 Ξ”
jax-v0.3.3 Released March 17, 2022
679 Ξ”
jax-v0.3.2 Released March 16, 2022
13,751 Ξ”
jax-v0.3.1 Released February 18, 2022
6,721 Ξ”

Top Contributors in jax-v0.3.6

mattjj
sharadmv
hawkinsp
a-googler
lgeiger
jakevdp
yashk2810
tlu7
jpuigcerver

Directory Browser for jax-v0.3.6

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

Release Notes Published

  • Changes:
    • Upgraded libtpu wheel to the fixed version. Fixes #10218.
GitClear uses cookies to ensure you get the best experience on our website. Learn more