README.md
setup.cfg
setup.py
examples/mnist.py
examples/mnist_dataset.py
examples/toy_model_tpu.py
mesh_tensorflow/__init__.py
mesh_tensorflow/beam_search.py
mesh_tensorflow/import_test.py
mesh_tensorflow/layers.py
mesh_tensorflow/layers_test.py
mesh_tensorflow/ops.py
mesh_tensorflow/ops_test.py
mesh_tensorflow/ops_with_redefined_builtins.py
mesh_tensorflow/optimize.py
mesh_tensorflow/placement_mesh_impl.py
mesh_tensorflow/simd_mesh_impl.py
mesh_tensorflow/simd_mesh_impl_test.py
mesh_tensorflow/tpu_variables.py
mesh_tensorflow/utils.py
mesh_tensorflow/utils_test.py
mesh_tensorflow.egg-info/PKG-INFO
mesh_tensorflow.egg-info/SOURCES.txt
mesh_tensorflow.egg-info/dependency_links.txt
mesh_tensorflow.egg-info/requires.txt
mesh_tensorflow.egg-info/top_level.txt
mesh_tensorflow/auto_mtf/__init__.py
mesh_tensorflow/auto_mtf/api.py
mesh_tensorflow/auto_mtf/api_test.py
mesh_tensorflow/auto_mtf/graph_interface.py
mesh_tensorflow/auto_mtf/graph_interface_test.py
mesh_tensorflow/auto_mtf/layout_optimizer.py
mesh_tensorflow/auto_mtf/layout_optimizer_test.py
mesh_tensorflow/auto_mtf/memory_estimator.py
mesh_tensorflow/auto_mtf/memory_estimator_test.py
mesh_tensorflow/auto_mtf/print_cp_model_solution.py
mesh_tensorflow/auto_mtf/scheduler.py
mesh_tensorflow/auto_mtf/scheduler_test.py
mesh_tensorflow/auto_mtf/valid_layouts.py
mesh_tensorflow/auto_mtf/valid_layouts_test.py
mesh_tensorflow/bert/__init__.py
mesh_tensorflow/bert/bert.py
mesh_tensorflow/bert/optimization.py
mesh_tensorflow/bert/run_classifier.py
mesh_tensorflow/bert/run_pretraining.py
mesh_tensorflow/bert/run_squad.py
mesh_tensorflow/bert/tokenization.py
mesh_tensorflow/experimental/__init__.py
mesh_tensorflow/experimental/data_aug_lib.py
mesh_tensorflow/experimental/data_aug_lib_test.py
mesh_tensorflow/experimental/input_reader.py
mesh_tensorflow/experimental/input_reader_test.py
mesh_tensorflow/experimental/model_executor.py
mesh_tensorflow/experimental/offline_data_aug.py
mesh_tensorflow/experimental/unet.py
mesh_tensorflow/transformer/__init__.py
mesh_tensorflow/transformer/attention.py
mesh_tensorflow/transformer/dataset.py
mesh_tensorflow/transformer/learning_rate_schedules.py
mesh_tensorflow/transformer/learning_rate_schedules_test.py
mesh_tensorflow/transformer/main.py
mesh_tensorflow/transformer/moe.py
mesh_tensorflow/transformer/t2t_vocabulary.py
mesh_tensorflow/transformer/transformer.py
mesh_tensorflow/transformer/transformer_layers.py
mesh_tensorflow/transformer/utils.py
mesh_tensorflow/transformer/utils_test.py
mesh_tensorflow/transformer/vocabulary.py
mesh_tensorflow/transformer/gin/8x8_dp_r4.gin
mesh_tensorflow/transformer/gin/__init__.py
mesh_tensorflow/transformer/gin/defaults.gin
mesh_tensorflow/transformer/gin/transparent_attention.gin