This section contains shorter notes on topics relevant to using JAX; see also the longer design discussions in JAX Enhancement Proposals (JEPs).
- Dependencies and version compatibility:
- Migrations and deprecations:
jax.Array migration summarizes the changes to the default array type in jax v 0.4.1
- Memory and computation usage:
- Programmer guardrails: