No public description

PiperOrigin-RevId: 619147712

Author: learned_optimization authors

docs/notebooks/summary_tutorial.ipynb

@@ -353,7 +353,7 @@
    "source": [
     "Then we can transform the `loss` function with the function transformation: `summary.with_summary_output_reduced`.\n",
     "This transformation goes through the computation and extracts all the tagged values and returns them to us by name in a dictionary.\n",
-    "In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/tensorflow/probability/blob/main/spinoffs/oryx/oryx/core/interpreters/harvest.py)).\n",
+    "In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/jax-ml/oryx/tree/main/oryx/core/interpreters/harvest.py)).\n",
     "When we wrap a function this, we return a tuple containing the original result, and a dictionary with the desired metrics."
    ]
   },

docs/notebooks/summary_tutorial.md

@@ -112,7 +112,7 @@ def loss(parameters):
 
 Then we can transform the `loss` function with the function transformation: `summary.with_summary_output_reduced`.
 This transformation goes through the computation and extracts all the tagged values and returns them to us by name in a dictionary.
-In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/tensorflow/probability/blob/main/spinoffs/oryx/oryx/core/interpreters/harvest.py)).
+In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/jax-ml/oryx/tree/main/oryx/core/interpreters/harvest.py)).
 When we wrap a function this, we return a tuple containing the original result, and a dictionary with the desired metrics.
 
 ```{code-cell}

docs/notebooks/summary_tutorial.py

@@ -25,6 +25,7 @@
 #   kernelspec:
 #     display_name: Python 3
 #     name: python3
+# pylint: disable=line-too-long
 # ---
 
 # + [markdown] id="ryqPvTKI19zH"
@@ -99,7 +100,7 @@ def loss(parameters):
 # + [markdown] id="AL9_xgfR4yPS"
 # Then we can transform the `loss` function with the function transformation: `summary.with_summary_output_reduced`.
 # This transformation goes through the computation and extracts all the tagged values and returns them to us by name in a dictionary.
-# In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/tensorflow/probability/blob/main/spinoffs/oryx/oryx/core/interpreters/harvest.py)).
+# In implementation, all the hard work here is done by the wonderful `oryx` library (in particular [harvest](https://github.com/jax-ml/oryx/tree/main/oryx/core/interpreters/harvest.py)).
 # When we wrap a function this, we return a tuple containing the original result, and a dictionary with the desired metrics.
 
 # + colab={"base_uri": "https://localhost:8080/"} id="hZQkB6Um8PI5" outputId="984e4f64-7562-48ae-ca68-ff4014037553"