JAX v0.8.2

v0.8.2

JAX v0.8.2

Changed 1
  • jax's Tracer no longer inherits from jax.Array at runtime, but jax.Array now uses a custom metaclass such that isinstance(x, Array) is true if an object x represents a traced Array
Removed 1
  • jax.experimental.si_vjp has been deleted; use jax.vjp instead
Deprecated 4
  • jax.lax.pvary has been deprecated; use jax.lax.pcast(..., to='varying') as the replacement
  • Complex arguments passed to jax.numpy.arange now result in a deprecation warning
  • From jax.core, the following symbols are newly deprecated: 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

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

View original

Upgraded? How did it go?

Discussion