LICENSE
README.md
pyproject.toml
src/universal_set_transformers.egg-info/PKG-INFO
src/universal_set_transformers.egg-info/SOURCES.txt
src/universal_set_transformers.egg-info/dependency_links.txt
src/universal_set_transformers.egg-info/requires.txt
src/universal_set_transformers.egg-info/top_level.txt
src/ust/__init__.py
src/ust/attention_attribution.py
src/ust/checkpointing.py
src/ust/utils.py
src/ust/api/attention.py
src/ust/api/checkpointing.py
src/ust/api/modules.py
src/ust/api/types.py
src/ust/modules/__init__.py
src/ust/modules/scaled_dot_product_attention.py
src/ust/modules/set_transformer.py
test/test_checkpointing.py
test/test_modules.py
test/test_scaled_dot_product_attention.py
test/test_set_transformers.py