Notes#
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:
API compatibility outlines JAX’s policies with regard to API compatibility across releases.
Python and NumPy version support policy outlines JAX’s policies with regard to compatibility with Python and NumPy.
- Migrations and deprecations:
jax.Array migration summarizes the changes to the default array type in jax v 0.4.1
- Memory and computation usage:
Asynchronous dispatch describes JAX’s asynchronous dispatch model.
Concurrency describes how JAX interacts with other Python concurrency.
GPU memory allocation describes how JAX interacts with memory allocation on GPU.
- Programmer guardrails:
Rank promotion warning describes how to configure
jax.numpy
to avoid implicit rank promotion.