From c4648e9b191199008bff260271949696cec70121 Mon Sep 17 00:00:00 2001 From: Jake Harmon Date: Wed, 4 Dec 2024 15:46:00 -0800 Subject: [PATCH] Update references to JAX's GitHub repo JAX has moved from https://github.com/google/jax to https://github.com/jax-ml/jax PiperOrigin-RevId: 702886818 --- oryx/examples/notebooks/a_tour_of_oryx.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/oryx/examples/notebooks/a_tour_of_oryx.ipynb b/oryx/examples/notebooks/a_tour_of_oryx.ipynb index db41f78..ba97072 100644 --- a/oryx/examples/notebooks/a_tour_of_oryx.ipynb +++ b/oryx/examples/notebooks/a_tour_of_oryx.ipynb @@ -93,7 +93,7 @@ "id": "F_n9c7K3xdKQ" }, "source": [ - "Oryx is an experimental library that extends [JAX](https://github.com/google/jax) to applications ranging from building and training complex neural networks to approximate Bayesian inference in deep generative models. Like JAX provides `jit`, `vmap`, and `grad`, Oryx provides a set of **composable function transformations** that enable writing simple code and transforming it to build complexity while staying completely interoperable with JAX.\n", + "Oryx is an experimental library that extends [JAX](https://github.com/jax-ml/jax) to applications ranging from building and training complex neural networks to approximate Bayesian inference in deep generative models. Like JAX provides `jit`, `vmap`, and `grad`, Oryx provides a set of **composable function transformations** that enable writing simple code and transforming it to build complexity while staying completely interoperable with JAX.\n", "\n", "JAX can only safely transform pure, functional code (i.e. code without side-effects). While pure code can be easier to write and reason about, \"impure\" code can often be more concise and more easily expressive.\n", "\n",