https://github.com/google/jax
Raw File
Tip revision: fe0f12e00e53fd9557723c6398797cc4514992f1 authored by Adam Paszke on 18 September 2023, 17:53:10 UTC
[Pallas] Wire up cost estimates in Mosaic params
Tip revision: fe0f12e
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