From d3ccf0a5c7940193467eeeabf9fb5830a642739c Mon Sep 17 00:00:00 2001 From: Matthew Johnson Date: Tue, 9 Jun 2020 07:35:07 -0700 Subject: [PATCH] fix typo in docs --- docs/notebooks/How_JAX_primitives_work.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/notebooks/How_JAX_primitives_work.ipynb b/docs/notebooks/How_JAX_primitives_work.ipynb index 0ebdd505da66..6611593457ca 100644 --- a/docs/notebooks/How_JAX_primitives_work.ipynb +++ b/docs/notebooks/How_JAX_primitives_work.ipynb @@ -1280,7 +1280,7 @@ " if not ad.is_undefined_primal(x):\n", " # This use of multiply_add is with a constant \"x\"\n", " assert ad.is_undefined_primal(y)\n", - " ct_y = ad.Zero(y.aval) if type(ct) is ad.Zero else multiply_add_prim(x, ct, lax.zeros_like_array(x)) + " ct_y = ad.Zero(y.aval) if type(ct) is ad.Zero else multiply_add_prim(x, ct, lax.zeros_like_array(x))\n", " res = None, ct_y, ct\n", " else:\n", " # This use of multiply_add is with a constant \"y\"\n",