LICENSE
README.md
setup.py
triton_transformer/__init__.py
triton_transformer/autoregressive_wrapper.py
triton_transformer/bmm.py
triton_transformer/cross_entropy.py
triton_transformer/dropout.py
triton_transformer/layernorm.py
triton_transformer/softmax.py
triton_transformer/transformer.py
triton_transformer/utils.py
triton_transformer.egg-info/PKG-INFO
triton_transformer.egg-info/SOURCES.txt
triton_transformer.egg-info/dependency_links.txt
triton_transformer.egg-info/requires.txt
triton_transformer.egg-info/top_level.txt