jax-ml/jax jax-v0.8.2

JAX v0.8.2

on GitHub

Deprecations

• jax.lax.pvary has been deprecated.
  Please use jax.lax.pcast(..., to='varying') as the replacement.
• Complex arguments passed to jax.numpy.arange now result in a
  deprecation warning, because the output is poorly-defined.
• From jax.core a number of symbols are newly deprecated including:
  call_impl, get_aval, mapped_aval, subjaxprs, set_current_trace,
  take_current_trace, traverse_jaxpr_params, unmapped_aval,
  AbstractToken, and TraceTag.
• All symbols in jax.interpreters.pxla are deprecated. These are
  primarily JAX internal APIs, and users should not rely on them.

Changes:

• jax's Tracer no longer inherits from jax.Array at runtime. However,
  jax.Array now uses a custom metaclass such isinstance(x, Array) is true
  if an object x represents a traced Array. Only some Tracers represent
  Arrays, so it is not correct for Tracer to inherit from Array.

  For the moment, during Python type checking, we continue to declare Tracer
  as a subclass of Array, however we expect to remove this in a future
  release.

• jax.experimental.si_vjp has been deleted.
  jax.vjp subsumes it's functionality.