ray/doc/source/raysgd/v2/examples/tensorflow_mnist_example.rst