From 687d91f5cba8feb4315e3c38cd91d09e3c2ce5f9 Mon Sep 17 00:00:00 2001
From: osbm <osmanfbayram@gmail.com>
Date: Sat, 26 Oct 2024 19:19:21 +0300
Subject: [PATCH] add jax

---
 common/python.nix | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/common/python.nix b/common/python.nix
index b271b23..47d92f2 100644
--- a/common/python.nix
+++ b/common/python.nix
@@ -10,6 +10,7 @@
       python312Packages.pandas
       python312Packages.seaborn
       python312Packages.matplotlib
+      python312Packages.jax
     ]))
     (pkgs.python311.withPackages (ppkgs: [
       python311Packages.pip
@@ -19,6 +20,7 @@
       python311Packages.pandas
       python311Packages.seaborn
       python311Packages.matplotlib
+      python311Packages.jax
     ]))
     pyenv
   ];