mirror of
https://github.com/ROCm/jax.git
synced 2025-04-14 10:56:06 +00:00
15 lines
609 B
ReStructuredText
15 lines
609 B
ReStructuredText
Concurrency
|
|
===========
|
|
|
|
JAX has limited support for Python concurrency.
|
|
|
|
Clients may call JAX APIs (e.g., :func:`~jax.jit` or :func:`~jax.grad`)
|
|
concurrently from separate Python threads.
|
|
|
|
It is not permitted to manipulate JAX trace values concurrently from multiple
|
|
threads. In other words, while it is permissible to call functions that use JAX
|
|
tracing (e.g., :func:`~jax.jit`) from multiple threads, you must not use
|
|
threading to manipulate JAX values inside the implementation of the function
|
|
`f` that is passed to :func:`~jax.jit`. The most likely outcome if you do this
|
|
is a mysterious error from JAX.
|