mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 19:06:07 +00:00
34 lines
1.1 KiB
ReStructuredText
34 lines
1.1 KiB
ReStructuredText
.. _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 |