From 01a91142c86c9d5012dcf482ee105aaaca313798 Mon Sep 17 00:00:00 2001
From: Mathis Gerdes <MathisGerdes@gmail.com>
Date: Thu, 26 Dec 2024 03:34:37 +0100
Subject: [PATCH] Update package meta-data.

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

diff --git a/pyproject.toml b/pyproject.toml
index 7ecc9b7..a980934 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -7,7 +7,7 @@ name = "jax_autovmap"
 authors = [
     { name="Mathis Gerdes", email="mathisgerdes@gmail.com" },
 ]
-description = "Automatically broadcast inputs by dynamically applying jax.vmap"
+description = "Automatically broadcast inputs by dynamically applying jax.vmap given un-batched argument dimensions."
 readme = "README.md"
 requires-python = ">=3.7"
 classifiers = [
@@ -16,12 +16,10 @@ classifiers = [
     "Operating System :: OS Independent",
 ]
 dependencies = [
-    'jax>=0.2.20',
-    'jaxlib>=0.1.69',
-    'numpy',
+    'jax>=0.4.0',
     'importlib-metadata; python_version<"3.8"',
 ]
 dynamic = ["version"]
 
 [tool.setuptools.dynamic]
-version = {attr = "jax_autovmap.__version__"}
\ No newline at end of file
+version = {attr = "jax_autovmap.__version__"}