Fix images to stay at top-level directory and a link.

PiperOrigin-RevId: 217548849

Author: Dustin Tran

README.md

@@ -95,7 +95,7 @@ with 10 units (corresponding to the 10 digit classes).
 The code consists of two parts, the first describing the mathematical
 operations, and the second describing the devices and tensor/computation layout.
 For the full example, see [`examples/mnist.py`](
-https://github.com/tensorflow/mesh/tree/master/mesh_tensorflow/examples/mnist.py).
+https://github.com/tensorflow/mesh/blob/master/examples/mnist.py).
 TODO(noam): verify that this code works.
 
 ```Python

examples/mnist.py

@@ -13,9 +13,9 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""Mnist using mesh-tensorflow and tf.Estimator.
+"""MNIST using Mesh TensorFlow and TF Estimator.
 
-This is an illustration of mesh-tensorflow, not a good model.
+This is an illustration, not a good model.
 """
 
 from __future__ import absolute_import

examples/toy_model_tpu.py

@@ -13,7 +13,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-"""A toy model using mesh-tensrflow."""
+"""A toy model using Mesh TensorFlow."""
 
 from __future__ import absolute_import
 from __future__ import division