From 3f155272fc9f0080ec67094d8ec2e67200cb4ab7 Mon Sep 17 00:00:00 2001
From: Brennan Saeta <saeta@google.com>
Date: Thu, 9 Dec 2021 16:40:47 -0800
Subject: [PATCH] Bump JAX requirements.

PiperOrigin-RevId: 415386111
---
 requirements-jax.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements-jax.txt b/requirements-jax.txt
index c7d25bde2..612b64b3b 100644
--- a/requirements-jax.txt
+++ b/requirements-jax.txt
@@ -1,2 +1,2 @@
-jax==0.2.22
-jaxlib==0.1.73
+jax>=0.2.22
+jaxlib>=0.1.73