mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 03:46:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

add jax.block_until_ready to docs and changelog

Browse Source 
also unrelatedly fix a couple of the uses of rst in changelog.md (though
many others remain)
This commit is contained in:
Matthew Johnson 2021-12-14 13:36:46 -08:00
parent d6223cd4f8
commit 0c68605bf1
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
CHANGELOG.md
View File

@ -21,6 +21,9 @@ PLEASE REMEMBER TO CHANGE THE '..main' WITH AN ACTUAL TAG in GITHUB LINK.
Additionally, added documentation for how to implement the old behavior Additionally, added documentation for how to implement the old behavior
using JAX custom AD APIs ({jax-issue}`#7839`). using JAX custom AD APIs ({jax-issue}`#7839`).
* New features:
* add `jax.block_until_ready` ({jax-issue}`#8941)
## jaxlib 0.1.76 (Unreleased) ## jaxlib 0.1.76 (Unreleased)
## jaxlib 0.1.75 (Dec 8, 2021) ## jaxlib 0.1.75 (Dec 8, 2021)

1
docs/jax.rst
View File

@ -101,6 +101,7 @@ Parallelization (:code:`pmap`)
.. autofunction:: device_put_replicated .. autofunction:: device_put_replicated
.. autofunction:: device_put_sharded .. autofunction:: device_put_sharded
.. autofunction:: device_get .. autofunction:: device_get
.. autofunction:: block_until_ready
.. autofunction:: default_backend .. autofunction:: default_backend
.. autofunction:: named_call .. autofunction:: named_call