From 0c68605bf1af90662472966657d7a489ff3329a2 Mon Sep 17 00:00:00 2001
From: Matthew Johnson <mattjj@google.com>
Date: Tue, 14 Dec 2021 13:36:46 -0800
Subject: [PATCH] add jax.block_until_ready to docs and changelog

also unrelatedly fix a couple of the uses of rst in changelog.md (though
many others remain)
---
 CHANGELOG.md | 3 +++
 docs/jax.rst | 1 +
 2 files changed, 4 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index b88a48224..c73cec789 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -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)
diff --git a/docs/jax.rst b/docs/jax.rst
index 80710994b..75a746cab 100644
--- a/docs/jax.rst
+++ b/docs/jax.rst
@@ -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