README.md
setup.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/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/transformer/__init__.py
mesh_tensorflow/transformer/moe.py
mesh_tensorflow/transformer/transformer.py
mesh_tensorflow/transformer/transformer_layers.py