Update pyproject.toml
Browse files- pyproject.toml +5 -5
pyproject.toml
CHANGED
|
@@ -35,8 +35,8 @@ dependencies = [
|
|
| 35 |
|
| 36 |
[tool.uv] # optional, but handy
|
| 37 |
override = [
|
| 38 |
-
# keep the JAX triplet in lockstep explicitly
|
| 39 |
-
"jax==0.
|
| 40 |
-
"jaxlib==0.
|
| 41 |
-
"jax-cuda12-plugin==0.
|
| 42 |
-
]
|
|
|
|
| 35 |
|
| 36 |
[tool.uv] # optional, but handy
|
| 37 |
override = [
|
| 38 |
+
# keep the JAX triplet in lockstep explicitly - updated for TF 2.20 compatibility
|
| 39 |
+
"jax==0.4.38",
|
| 40 |
+
"jaxlib==0.4.38",
|
| 41 |
+
"jax-cuda12-plugin==0.4.38",
|
| 42 |
+
]
|