From bdf07a79a251101b728b8f3c2670e3384e10d959 Mon Sep 17 00:00:00 2001
From: singularitti <singularitti@outlook.com>
Date: Sat, 17 Feb 2024 20:15:26 -0500
Subject: [PATCH] Add `jax` to deps

---
 pyproject.toml | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pyproject.toml b/pyproject.toml
index 53cfb6c..8ec143e 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -15,6 +15,7 @@ dependencies = [
     "pyyaml>=6.0.1",
     "matplotlib>=3.8.2",
     "numba>=0.58.1",
+    "jax>=0.4.24",
 ]
 requires-python = ">=3.11"
 readme = "README.md"