From f73753cf7af1fb5a4b302d089644fe49067735cd Mon Sep 17 00:00:00 2001
From: Mike Anderson <mikeanderson@google.com>
Date: Fri, 28 Oct 2022 13:54:12 -0700
Subject: [PATCH] Internal change

PiperOrigin-RevId: 484613416
Change-Id: I258cbe06a1e1b5afca7388beafbe114e72926a99
---
 requirements/requirements.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements/requirements.txt b/requirements/requirements.txt
index 6d56183..9f191cb 100644
--- a/requirements/requirements.txt
+++ b/requirements/requirements.txt
@@ -16,8 +16,8 @@ absl-py
 arviz==0.11.2
 dataclasses;python_version<'3.7'
 immutabledict>=2.0.0
-jax>=0.3.14
-jaxlib>=0.3.14
+jax>=0.3.18
+jaxlib>=0.3.18
 matplotlib==3.3.4
 numpy>=1.12
 numpyro>=0.9.2