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__"}