- GitHub commits.
- Changes
- Ahead-of-time lowering and compilation functionality (tracked in
{jax-issue}#7733
) is stable and public. See the
overview and the API docs
for {mod}jax.stages
. - Introduced {class}
jax.Array
, intended to be used for bothisinstance
checks
and type annotations for array types in JAX. Notice that this included some subtle
changes to howisinstance
works for {class}jax.numpy.ndarray
for jax-internal
objects, as {class}jax.numpy.ndarray
is now a simple alias of {class}jax.Array
.
- Ahead-of-time lowering and compilation functionality (tracked in
- Breaking changes
jax._src
is no longer imported into the from the publicjax
namespace.
This may break users that were using JAX internals.jax.soft_pmap
has been deleted. Please usepjit
orxmap
instead.
jax.soft_pmap
is undocumented. If it were documented, a deprecation period
would have been provided.