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

Jax: jax-v0.3.7 Release

Release date:
April 15, 2022
Previous version:
jax-v0.3.6 (released April 13, 2022)
Magnitude:
4,073 Diff Delta
Contributors:
15 total committers
Data confidence:
Commits:
44 impacting commits

44 Commits in this Release

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

[MHLO] Change jax.xla_computation() to use MHLO lowering internally.
Authored April 13, 2022
Deprecate jax.experimental.loops, step 2.
Authored April 14, 2022
Add tests for higher order primitives
Authored April 12, 2022
add some simple iree tests
Authored April 13, 2022
[MHLO] Separate registrations for collective and initial_style primitives from the XLA translation rule registration.
Authored April 13, 2022
[MHLO] Add MHLO lowerings for FFT ops.
Authored April 14, 2022
[MHLO] Add MHLO lowerings for triangular_solve, cholesky, and schur.
Authored April 14, 2022
Expose the input and output sharding on the compiled object.
Authored April 13, 2022
Drop dead code now that the minimum jaxlib version is 0.3.2.
Authored April 13, 2022
better errors: check for callability of lax.control_flow arguments
Authored April 13, 2022
Enable context manager name stack with `jax2tf`
Authored April 7, 2022
Add polydiv to jax.numpy
Authored April 11, 2022
Bump minimum jaxlib version to 0.3.2 and remove transfer guard compatibility code
Authored April 11, 2022
debug_nans: don't return results of successfully running de-optimized function
Authored April 12, 2022
Deprecate DeviceArray.tile method
Authored April 13, 2022
Do not generate trivial gathers when indexing entire axis
Authored April 9, 2022
[MHLO] Add an MHLO lowering for rng_bit_generator.
Authored April 14, 2022
Avoid broadcasting the input and indices in jnp.take_along_axis.
Authored April 14, 2022
Expose CopyToRemoteDevice and MakeCrossHostReceiveBuffer in Python bindings.
Authored April 14, 2022
Fix PRNGKeyArray.broadcast_to with scalar shape
Authored April 14, 2022
Jax 0.3.7 and jaxlib 0.3.7 release.
Authored April 15, 2022
Clarify tree_flatten docstring.
Authored February 21, 2022
Fix#10219
Authored April 11, 2022
Remove use of xla.lower_fun in SVD translation rule.
Authored April 13, 2022
Fix#10247
Authored April 12, 2022
add default values to config context managers
Authored April 12, 2022
Finish jax release
Authored April 13, 2022
[MHLO] Add MHLO lowerings for name_p and unreachable_p.
Authored April 14, 2022
Skip remote_transfer_test on cloud TPU.
Authored April 15, 2022
Fix polydiv kokoro tests
Authored April 13, 2022
Comment on implementation of unzip2 & unzip3
Authored April 14, 2022
Add libstdc++ workaround for conda users.
Authored April 13, 2022
Fix arguments to schur translation rule
Authored April 13, 2022
[sparse] gate gpu lowering test to gpu backends
Authored April 14, 2022
Remove mention of six package from jaxlib build instructions.
Authored April 13, 2022
2nd attempt at jax/jaxlib 0.3.7 release.
Authored April 15, 2022
[jax2tf] Fixes conv bug. The filter dims are allowed to be bigger than the input dims if the padding type is "SAME".
Authored April 13, 2022
Improve precision of chlo.sinh.
Authored April 14, 2022
Temporarily disable integer index check in jnp.take_along_axis.
Authored April 15, 2022
Remove unused comments related to Python 2 compatibility.
Authored April 14, 2022

Browse Other Releases

Latest Pending Unreleased 😎
jax-v0.3.11 Released May 14, 2022
9,163 Ξ”
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 Ξ”

Top Contributors in jax-v0.3.7

hawkinsp
sharadmv
mattjj
gnecula
jakevdp
yashk2810
YouJiacheng
ntlm1686
hyeontaek
lgeiger

Directory Browser for jax-v0.3.7

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

Release Notes Published

  • Fixed a performance problem if the indices passed to jax.numpy.take_along_axis were broadcasted (#10281).
  • jax.scipy.special.expit and jax.scipy.special.logit now require their arguments to be scalars or JAX arrays. They also now promote integer arguments to floating point.
  • The DeviceArray.tile() method is deprecated, because numpy arrays do not have a tile() method. As a replacement for this, use jax.numpy.tile (#10266).
GitClear uses cookies to ensure you get the best experience on our website. Learn more