1
0
mirror of https://github.com/ROCm/jax.git synced 2025-04-20 22:06:06 +00:00

9 Commits

Author SHA1 Message Date
Roy Frostig
0bdbe763aa update documentation notes to new-style typed keys 2024-03-07 12:40:09 -08:00
ttttanya
9e50b9dc80 Asynchronous dispatch doc update regarding jax.Array migration 2023-10-05 13:45:51 -07:00
yashkatariya
ca6465dded Add the changes suggested 2021-08-01 20:13:02 -07:00
yashkatariya
be082b7614 Add some clarification 2021-07-29 06:50:42 -07:00
Jake VanderPlas
d85d204897 DOC: change build from nbsphinx to myst 2021-02-16 10:28:39 -08:00
Jake Vanderplas
05904faf0f
Change onp/np to np/jnp in docs & notebooks () 2020-07-15 13:17:38 -07:00
Jamie Townsend
670fab59cf
Test code in docs and api.py docstrings ()
Also remove jaxpr doc tests from api_test.py.
2020-05-16 16:19:24 +03:00
Stephan Hoyer
c3bf532aae Fix sphinx references in JAX docs
The links in `concurrency.rst` should work now.

The links in async_dispatch.rst don't work yet, but they will if
``DeviceArray`` ever ends up JAX's API docs.
2019-08-10 16:40:08 -07:00
Peter Hawkins
91e6ba322c Add documentation on asynchronous dispatch. 2019-06-04 10:09:43 -04:00