LICENSE
README.md
pyproject.toml
snax/__init__.py
snax/attention.py
snax/base.py
snax/checkpoint.py
snax/conditional.py
snax/dataset.py
snax/made.py
snax/nn.py
snax/recurrent.py
snax/test_attention.py
snax/test_checkpoint.py
snax/test_dataset.py
snax/test_made.py
snax/test_recurrent.py
snax/test_train_lib.py
snax/test_utils.py
snax/test_vmapped.py
snax/train_lib.py
snax/utils.py
snax/old/base.py
snax/old/conditional.py
snax/old/nn.py
snax/old/recurrent.py
snaxlib.egg-info/PKG-INFO
snaxlib.egg-info/SOURCES.txt
snaxlib.egg-info/dependency_links.txt
snaxlib.egg-info/requires.txt
snaxlib.egg-info/top_level.txt