fix typo (#1652)

ShoufaChen · holly1238 · web-flow · commit 011ae8a6d47a · 2021-08-31T09:10:15.000-07:00
Co-authored-by: Holly Sweeney &lt;77758406+holly1238@users.noreply.github.com&gt;
diff --git a/prototype_source/vmap_recipe.py b/prototype_source/vmap_recipe.py
@@ -27,8 +27,9 @@
 # that runs on examples and then lift it to a function that can
 # take batches of examples with `vmap(func)`. `func` however
 # is subject to many restrictions:
+#
 # - it must be functional (one cannot mutate a Python data structure
-#   inside of it), with teh exception of in-place PyTorch operations.
+#   inside of it), with the exception of in-place PyTorch operations.
 # - batches of examples must be provided as Tensors. This means that
 #   vmap doesn't handle variable-length sequences out of the box.
 #
@@ -67,6 +68,7 @@ def model(feature_vec):
 # `vmap` can also help vectorize computations that were previously difficult
 # or impossible to batch. This bring us to our second use case: batched
 # gradient computation.
+#
 # - https://github.com/pytorch/pytorch/issues/8304
 # - https://github.com/pytorch/pytorch/issues/23475
 #
