JAX is no longer a hard dependency (to support e.g. PyTorch) (#50)

2026-09-09 11:24:55 +08:00 · 2022-12-07 10:48:47 -08:00
parent 8fbf7bf3a5
commit 29654e7087
3 changed files with 28 additions and 18 deletions
@@ -71,7 +71,6 @@ python_requires = "~=3.7"
 # https://github.com/explosion/confection/blob/main/setup.cfg#L33 used in colab

 install_requires = [
-    "jax>=0.3.4",
    "numpy>=1.20.0",
    "typeguard>=2.13.3",
    "typing_extensions>=3.7.4.1",