From 846e82723f7da5d977c560e58775261eba585286 Mon Sep 17 00:00:00 2001 From: Martin Schubert Date: Mon, 14 Oct 2024 14:32:52 -0700 Subject: [PATCH] pin jax --- pyproject.toml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pyproject.toml b/pyproject.toml index 506c8c2..cc6b695 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -19,7 +19,7 @@ dependencies = [ "agjax", "ceviche_challenges", "fmmax", - "jax", + "jax <= 0.4.34", "jaxlib", "numpy", "refractiveindex",