pin nccl version

Ruixuan · changlan · commit 2e03b1229115 · 2025-07-25T09:59:55.000-07:00
* pin nccl version * empty commit * add actual pacakge * trigger new build to address flaky test * Update pyproject.toml GitOrigin-RevId: b8653ad
diff --git a/pyproject.toml b/pyproject.toml
@@ -137,6 +137,8 @@ gpu = [
     "triton==2.1.0",
     "jax[cuda12]==0.5.3",
     "nvidia-ml-py==12.560.30",
+    # pin nccl version, otherwise jax[cuda12] will pull latest version
+    "nvidia-nccl-cu12==2.27.5",
 ]
 # Open API inference.
 open_api = [

Original file line number	Diff line number	Diff line change
`@@ -137,6 +137,8 @@ gpu = [`
`137`	`137`	`"triton==2.1.0",`
`138`	`138`	`"jax[cuda12]==0.5.3",`
`139`	`139`	`"nvidia-ml-py==12.560.30",`
	`140`	`+ # pin nccl version, otherwise jax[cuda12] will pull latest version`
	`141`	`+ "nvidia-nccl-cu12==2.27.5",`
`140`	`142`	`]`
`141`	`143`	`# Open API inference.`
`142`	`144`	`open_api = [`