[shard_map docs]: Fix doc typos

PiperOrigin-RevId: 656265100
jax-ml · Jul 26, 2024 · 8ed94bc · 8ed94bc
1 parent 0f834cd
commit 8ed94bc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/docs/notebooks/shard_map.ipynb b/docs/notebooks/shard_map.ipynb
@@ -1129,7 +1129,7 @@
    "id": "3e76bcfd",
    "metadata": {},
    "source": [
-    "And wee shard the right-hand side argument (can think: activations) on its\n",
+    "And we shard the right-hand side argument (can think: activations) on its\n",
     "contracting dimension, with a similar sharding for the output:"
    ]
   },

diff --git a/docs/notebooks/shard_map.md b/docs/notebooks/shard_map.md
@@ -820,7 +820,7 @@ lhs_spec = P('i', None)
 lhs = device_put(jax.random.normal(jax.random.key(0), (8, 8)), lhs_spec)
 ```
 
-And wee shard the right-hand side argument (can think: activations) on its
+And we shard the right-hand side argument (can think: activations) on its
 contracting dimension, with a similar sharding for the output:
 
 ```{code-cell}