diff --git a/docs/notebooks/shard_map.ipynb b/docs/notebooks/shard_map.ipynb index ed0a13d8702a..919690d230ab 100644 --- 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 index 67494cfd4a02..6f9dfbb659e1 100644 --- 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}