LICENSE
README.md
setup.py
scalax/__init__.py
scalax/attention.py
scalax/sharding.py
scalax/utils.py
scalax.egg-info/PKG-INFO
scalax.egg-info/SOURCES.txt
scalax.egg-info/dependency_links.txt
scalax.egg-info/requires.txt
scalax.egg-info/top_level.txt