.editorconfig
.gitattributes
.gitignore
.pre-commit-config.yaml
CODE_OF_CONDUCT.md
LICENSE
README.md
SECURITY.md
SUPPORT.md
azure-pipelines.yml
environment.yml
pyproject.toml
setup.py
.github/workflows/codeql-analysis.yml
.github/workflows/pre-commit.yml
.github/workflows/publish-docs.yml
.github/workflows/publish.yml
.github/workflows/tests.yml
docs/Makefile
docs/conf.py
docs/index.rst
docs/modules/relationalmha.rst
docs/modules/relationaltransformer.rst
docs/modules/settransformers.rst
docs/modules/utils.rst
mi_module_zoo/__init__.py
mi_module_zoo/__version__.py
mi_module_zoo/mlp.py
mi_module_zoo/py.typed
mi_module_zoo/relationalmultiheadattention.py
mi_module_zoo/relationaltransformerlayers.py
mi_module_zoo/settransformer.py
mi_module_zoo.egg-info/PKG-INFO
mi_module_zoo.egg-info/SOURCES.txt
mi_module_zoo.egg-info/dependency_links.txt
mi_module_zoo.egg-info/not-zip-safe
mi_module_zoo.egg-info/requires.txt
mi_module_zoo.egg-info/top_level.txt
mi_module_zoo/utils/__init__.py
mi_module_zoo/utils/activation.py
mi_module_zoo/utils/random.py
mi_module_zoo/utils/testing.py
tests/__init__.py
tests/test_activations.py
tests/test_mlp.py
tests/test_relationalmultiheadattention.py
tests/test_relationaltransformers.py
tests/test_settransformers.py