mirror of
https://github.com/ROCm/jax.git
synced 2025-04-19 05:16:06 +00:00
37 lines
1.2 KiB
ReStructuredText
37 lines
1.2 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.
|
|
|
|
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.
|
|
|
|
Arrays and data types:
|
|
- :doc:`type_promotion` describes JAX's implicit type promotion for functions of two or more values.
|
|
- :doc:`default_dtypes` describes how JAX determines the default dtype for array creation functions.
|
|
|
|
|
|
.. toctree::
|
|
:hidden:
|
|
:maxdepth: 1
|
|
|
|
api_compatibility
|
|
deprecation
|
|
async_dispatch
|
|
concurrency
|
|
gpu_memory_allocation
|
|
rank_promotion_warning
|
|
type_promotion
|
|
default_dtypes
|