update jaxlib version and changelog for pypi (#3824)

WORKSPACE

@@ -28,10 +28,10 @@ http_archive(
 #    and update the sha256 with the result.
 http_archive(
     name = "org_tensorflow",
-    sha256 = "6a21839b7df5b06799a1c6190b1e7a4997817e7c88b35581de5f4b53b3885155",
-    strip_prefix = "tensorflow-8ed58f3448c6c517f83ead72a9cde9322cb2952c",
+    sha256 = "829ba588007670d9ff72d45f7ed20dbe4a5a7ff03b1cd89cd07d942aca2cf848",
+    strip_prefix = "tensorflow-a225c7700807350cf022b8256ed36b01acefe4d9",
     urls = [
-        "https://github.com/tensorflow/tensorflow/archive/8ed58f3448c6c517f83ead72a9cde9322cb2952c.tar.gz",
+        "https://github.com/tensorflow/tensorflow/archive/a225c7700807350cf022b8256ed36b01acefe4d9.tar.gz",
     ],
 )
 

docs/CHANGELOG.rst

@@ -9,18 +9,25 @@ Change Log
 
 These are the release notes for JAX.
 
+jaxlib 0.1.52 (July 22, 2020)
+------------------------------
+
+* Update XLA.
+
 jax 0.1.73 (July 22, 2020)
 --------------------------
 * `GitHub commits <https://github.com/google/jax/compare/jax-v0.1.72...jax-v0.1.73>`_.
 * The minimum jaxlib version is now 0.1.51.
 
 * New Features:
+
   * jax.image.resize. (#3703)
   * hfft and ihfft (#3664)
   * jax.numpy.intersect1d (#3726)
   * jax.numpy.lexsort (#3812)
 
 * Bug Fixes:
+
   * Fix reduction repeated axis error (#3618)
   * Fix shape rule for lax.pad for input dimensions of size 0. (#3608)
   * make psum transpose handle zero cotangents (#3653)
@@ -29,6 +36,7 @@ jax 0.1.73 (July 22, 2020)
   * address nan issue in jax.scipy.special.zeta (#3777)
 
 * Improvements:
+
   * Many improvements to jax2tf
   * Reimplement argmin/argmax using a single pass variadic reduction. (#3611)
   * Enable XLA SPMD partitioning by default. (#3151)
@@ -42,7 +50,6 @@ jax 0.1.73 (July 22, 2020)
   * Enable buffer donation for GPU. (#3800)
   * Add support for base dilation and window dilation to reduce window op… (#3803)
 
-
 jaxlib 0.1.51 (July 2, 2020)
 ------------------------------
 

jaxlib/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "0.1.51"
+__version__ = "0.1.52"