Jax: jax-v0.3.7 Release

Release date:
April 15, 2022
Previous version:
jax-v0.3.6 (released April 13, 2022)
Magnitude:
506 Diff Delta
Contributors:
6 total committers
Data confidence:
Commits:

Top Contributors in jax-v0.3.7

sharadmv
ntlm1686
hyeontaek
lgeiger
GJBoth
YouJiacheng

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).