jax-ml/jax jax-v0.3.5

JAX release v0.3.5

on GitHub

Changes

• added jax.random.loggamma & improved behavior of jax.random.beta
  and jax.random.dirichlet for small parameter values (#9906).
• the private lax_numpy submodule is no longer exposed in the jax.numpy namespace (#10029).
• added array creation routines jax.numpy.frombuffer, jax.numpy.fromfunction,
  and jax.numpy.fromstring (#10049).
• DeviceArray.copy() now returns a DeviceArray rather than a np.ndarray (#10069)
• added jax.scipy.linalg.rsf2csf
• Deprecations:
  • jax.nn.normalize is being deprecated. Use jax.nn.standardize instead (#9899).
  • jax.tree_util.tree_multimap is deprecated. Use jax.tree_util.tree_map instead (#5746).
  • jax.experimental.sharded_jit is deprecated. Use pjit instead.