From 6bc5e9c6bf2674aad2ac9893135a145ef9884d41 Mon Sep 17 00:00:00 2001
From: Priya Kasimbeg <kasimbeg@google.com>
Date: Tue, 26 Mar 2024 02:10:29 +0000
Subject: [PATCH] fix protobuf version

---
 setup.cfg | 1 +
 1 file changed, 1 insertion(+)

diff --git a/setup.cfg b/setup.cfg
index 0c986451b..321020ad9 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -120,6 +120,7 @@ jax_core_deps =
   # upgrade jax.
   chex==0.1.7
   ml_dtypes==0.2.0
+  protobuf==4.25.3
 
 # JAX CPU
 jax_cpu =