Skip to content

Commit ef886c7

Browse files
gneculajekbradbury
andauthored
Update docs/notebooks/How_JAX_primitives_work.ipynb
Co-Authored-By: James Bradbury <jekbradbury@google.com>
1 parent 7bc0f70 commit ef886c7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/notebooks/How_JAX_primitives_work.ipynb

+2-2
Original file line numberDiff line numberDiff line change
@@ -1515,7 +1515,7 @@
15151515
"colab_type": "text"
15161516
},
15171517
"source": [
1518-
"We need to tell JAX how to evaluate the batched version of the primitive. In this particular case, the `multiply_add_prim` already operates pointwise for any dimension of input vectors. So the batched version can use the same `multiple_add_prim` implementation."
1518+
"We need to tell JAX how to evaluate the batched version of the primitive. In this particular case, the `multiply_add_prim` already operates pointwise for any dimension of input vectors. So the batched version can use the same `multiply_add_prim` implementation."
15191519
]
15201520
},
15211521
{
@@ -1648,4 +1648,4 @@
16481648
]
16491649
}
16501650
]
1651-
}
1651+
}

0 commit comments

Comments
 (0)