From 1e7bf5547d778c245c81fbc281d3c8b2a9df5bf0 Mon Sep 17 00:00:00 2001
From: Peter Hawkins <phawkins@google.com>
Date: Mon, 16 Nov 2020 12:15:34 -0500
Subject: [PATCH] Update XLA.

---
 WORKSPACE          | 6 +++---
 docs/CHANGELOG.rst | 2 ++
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/WORKSPACE b/WORKSPACE
index 8ac1d62f940a..297cdab35378 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -28,10 +28,10 @@ http_archive(
 #    and update the sha256 with the result.
 http_archive(
     name = "org_tensorflow",
-    sha256 = "34f473fc2dc80ad375a196bf6e98812c65d2eeb39aed6521aa0b2b03d34eb520",
-    strip_prefix = "tensorflow-4cf039a9877a5e7a73cf3bc189af0bd03d2cb422",
+    sha256 = "579a74ad171d8da7b7193ff863f28482c2e6050c4090650b001fb80bbc46bb0f",
+    strip_prefix = "tensorflow-04f25b55e27be95ec340f414c2a1cabe16be5c2a",
     urls = [
-        "https://github.com/tensorflow/tensorflow/archive/4cf039a9877a5e7a73cf3bc189af0bd03d2cb422.tar.gz",
+        "https://github.com/tensorflow/tensorflow/archive/04f25b55e27be95ec340f414c2a1cabe16be5c2a.tar.gz",
     ],
 )
 
diff --git a/docs/CHANGELOG.rst b/docs/CHANGELOG.rst
index 9da229f4788e..c1fe460fde89 100644
--- a/docs/CHANGELOG.rst
+++ b/docs/CHANGELOG.rst
@@ -54,6 +54,8 @@ jax 0.2.6 (Unreleased)
 jaxlib 0.1.58 (Unreleased)
 ------------------------------
 
+* Fixed a bug that meant JAX sometimes return platform-specific types (e.g.,
+  `np.cint`) instead of standard types (e.g., `np.int32`). (#4903)
 
 jaxlib 0.1.57 (November 12 2021)
 ------------------------------