mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36: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
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.75 (Dec 8, 2021)

1
docs/jax.rst
View File

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