Jax release v0.2.24

yashk2810 released this 19 Oct 15:06

· 15846 commits to main since this release

a7c9b6d

New features:
- jax.random.choice and jax.random.permutation now support
  multidimensional arrays and an optional axis argument (#8158)
Breaking changes:
- jax.numpy.take and jax.numpy.take_along_axis now require array-like inputs
  (see #7737)

Assets 2