From 8ed94bcfb6ab02e1f5fcaedb0190cf21640efc72 Mon Sep 17 00:00:00 2001 From: jax authors Date: Thu, 25 Jul 2024 23:29:17 -0700 Subject: [PATCH] [shard_map docs]: Fix doc typos PiperOrigin-RevId: 656265100 --- docs/notebooks/shard_map.ipynb | 2 +- docs/notebooks/shard_map.md | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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}