README.md
setup.py
tensorflow_modules/__init__.py
tensorflow_modules.egg-info/PKG-INFO
tensorflow_modules.egg-info/SOURCES.txt
tensorflow_modules.egg-info/dependency_links.txt
tensorflow_modules.egg-info/requires.txt
tensorflow_modules.egg-info/top_level.txt
tensorflow_modules/mask/__init__.py
tensorflow_modules/mask/util.py
tensorflow_modules/multi_head_attention/__init__.py
tensorflow_modules/multi_head_attention/attention.py
tensorflow_modules/multi_head_attention/model.py
tensorflow_modules/reversible_transformer/__init__.py
tensorflow_modules/reversible_transformer/reversible_decoder_layer.py
tensorflow_modules/reversible_transformer/reversible_encoder_layer.py
tensorflow_modules/transformer/__init__.py
tensorflow_modules/transformer/decoder_layer.py
tensorflow_modules/transformer/encoder_layer.py
tensorflow_modules/transformer/point_wise_feed_forward_network.py
tests/__init__.py
tests/mask/__init__.py
tests/mask/test_util.py
tests/multi_head_attention/__init__.py
tests/multi_head_attention/test_attention.py
tests/reversible_transformer/__init__.py
tests/reversible_transformer/test_reversible_decoder_layer.py
tests/reversible_transformer/test_reversible_encoder_layer.py
tests/transformer/__init__.py
tests/transformer/test_decoder_layer.py
tests/transformer/test_encoder_layer.py