2019-08-09 15:08:26 -04:00
|
|
|
Concurrency
|
|
|
|
|
===========
|
|
|
|
|
|
|
|
|
|
JAX has some limited support for Python concurrency.
|
|
|
|
|
|
|
|
|
|
Concurrency support is experimental and only lightly tested; please report any
|
|
|
|
|
bugs.
|
|
|
|
|
|
2019-08-10 16:40:08 -07:00
|
|
|
Clients may call JAX APIs (e.g., :func:`~jax.jit` or :func:`~jax.grad`)
|
|
|
|
|
concurrently from separate Python threads.
|
2019-08-09 15:08:26 -04:00
|
|
|
|
|
|
|
|
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
|
2019-08-10 16:40:08 -07:00
|
|
|
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.
|
