From 54dfa581f888c28e128a2d12de61e6ad48a18b81 Mon Sep 17 00:00:00 2001
From: yjwang <72957567+wangyji@users.noreply.github.com>
Date: Thu, 25 Jul 2024 22:12:44 -0400
Subject: [PATCH] Update pyproject.toml

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

diff --git a/pyproject.toml b/pyproject.toml
index 0cbc9ec..a3561c1 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -13,6 +13,14 @@ classifiers = [
     "Operating System :: OS Independent",
 ]
 
+dependencies = [
+  "jax >= 0.4.20",
+  "jaxlib >= 0.4.20",
+  "tensorflow-probability >= 0.23.0",
+  "matplotlib >= 3.8.0",
+  "optax >= 0.1.7"
+]
+
 [project.urls]
 Homepage = "https://github.com/YaoGroup/DIFFICE_jax"
 Issues = "https://github.com/YaoGroup/DIFFICE_jax/issues"