Update references to JAX's GitHub repo

jakeharmon8 · copybara-github · commit cad430fc22f5 · 2025-03-05T19:06:15.000-08:00
JAX has moved from https://github.com/google/jax to https://github.com/jax-ml/jax PiperOrigin-RevId: 702886625
diff --git a/agent_based_epidemic_sim/learning/covid_risk_score_sim_learning.ipynb b/agent_based_epidemic_sim/learning/covid_risk_score_sim_learning.ipynb
@@ -3116,7 +3116,7 @@
           "name": "stderr",
           "output_type": "stream",
           "text": [
-            "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/google/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
+            "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/jax-ml/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
           ]
         },
         {
@@ -3986,7 +3986,7 @@
           "name": "stderr",
           "output_type": "stream",
           "text": [
-            "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/google/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
+            "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/jax-ml/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
           ]
         },
         {
@@ -4169,7 +4169,7 @@
           "name": "stderr",
           "output_type": "stream",
           "text": [
-            "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/google/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
+            "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/jax-ml/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
           ]
         },
         {
@@ -10498,7 +10498,7 @@
           "name": "stderr",
           "output_type": "stream",
           "text": [
-            "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/google/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
+            "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/jax-ml/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
           ]
         },
         {

Original file line number	Diff line number	Diff line change
`@@ -3116,7 +3116,7 @@`
`3116`	`3116`	`"name": "stderr",`
`3117`	`3117`	`"output_type": "stream",`
`3118`	`3118`	`"text": [`
`3119`		- "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/google/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
	`3119`	+ "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/jax-ml/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
`3120`	`3120`	`]`
`3121`	`3121`	`},`
`3122`	`3122`	`{`
`@@ -3986,7 +3986,7 @@`
`3986`	`3986`	`"name": "stderr",`
`3987`	`3987`	`"output_type": "stream",`
`3988`	`3988`	`"text": [`
`3989`		- "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/google/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
	`3989`	+ "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/jax-ml/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
`3990`	`3990`	`]`
`3991`	`3991`	`},`
`3992`	`3992`	`{`
`@@ -4169,7 +4169,7 @@`
`4169`	`4169`	`"name": "stderr",`
`4170`	`4170`	`"output_type": "stream",`
`4171`	`4171`	`"text": [`
`4172`		- "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/google/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
	`4172`	+ "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/jax-ml/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
`4173`	`4173`	`]`
`4174`	`4174`	`},`
`4175`	`4175`	`{`
`@@ -10498,7 +10498,7 @@`
`10498`	`10498`	`"name": "stderr",`
`10499`	`10499`	`"output_type": "stream",`
`10500`	`10500`	`"text": [`
`10501`		- "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/google/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
	`10501`	+ "/usr/local/lib/python3.6/dist-packages/ipykernel_launcher.py:77: FutureWarning: Using a non-tuple sequence for multidimensional indexing is deprecated; use `arr[array(seq)]` instead of `arr[seq]`. In the future this will result in a TypeError. See https://github.com/jax-ml/jax/issues/4564 for discussion of why this type of indexing is being deprecated.\n"
`10502`	`10502`	`]`
`10503`	`10503`	`},`
`10504`	`10504`	`{`