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.