https://github.com/google/jax
Raw File
Tip revision: 500da57e91eb4b9094eb2b5cfa0826691862417f authored by jax authors on 07 May 2024, 14:07:04 UTC
Merge pull request #21077 from merrymercy:patch-1
Tip revision: 500da57
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