We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 61a7889 commit 54b518cCopy full SHA for 54b518c
2 files changed
pyproject.toml
@@ -52,9 +52,9 @@ gpu-cu12 = ["cupy-cuda12x"]
52
gpu-cu13 = ["cupy-cuda13x"]
53
stat = ["pytensor", "pymc"]
54
deep = ["torch", "jax"]
55
-deep-cu126 = ["torch", "jax[cuda12]"]
56
-deep-cu128 = ["torch", "jax[cuda12]"]
57
-deep-cu13 = ["torch>=2.11", "jax[cuda13]"]
+deep-cu126 = ["torch<2.11.0", "jax[cuda12]"]
+deep-cu128 = ["torch<2.11.0", "jax[cuda12]"]
+deep-cu13 = ["torch>=2.11.0", "jax[cuda13]"]
58
59
[dependency-groups]
60
dev = [
@@ -101,6 +101,7 @@ conflicts = [
101
{ extra = "deep" },
102
{ extra = "deep-cu126" },
103
{ extra = "deep-cu128" },
104
+ { extra = "deep-cu13" },
105
],
106
]
107
0 commit comments