https://github.com/google/jax
Raw File
Tip revision: fed7efd73003988282744b2f649df493b808c781 authored by jax authors on 03 April 2024, 18:45:56 UTC
Merge pull request #20571 from hawkinsp:release
Tip revision: fed7efd
notes.rst
.. _notes:

Notes
-----
This section contains shorter notes on topics relevant to using JAX; see also the
longer design discussions in :doc:`jep/index`.

Dependencies and version compatibility:
  - :doc:`api_compatibility` outlines JAX's policies with regard to API compatibility across releases.
  - :doc:`deprecation` outlines JAX's policies with regard to compatibility with Python and NumPy.

Migrations and deprecations:
  - :doc:`jax_array_migration` summarizes the changes to the default array type in jax v 0.4.1

Memory and computation usage:
  - :doc:`async_dispatch` describes JAX's asynchronous dispatch model.
  - :doc:`concurrency` describes how JAX interacts with other Python concurrency.
  - :doc:`gpu_memory_allocation` describes how JAX interacts with memory allocation on GPU.

Programmer guardrails:
  - :doc:`rank_promotion_warning` describes how to configure :mod:`jax.numpy` to avoid implicit rank promotion.


.. toctree::
   :hidden:
   :maxdepth: 1

   api_compatibility
   deprecation
   jax_array_migration
   async_dispatch
   concurrency
   gpu_memory_allocation
   rank_promotion_warning
back to top