LICENSE
README.md
pyproject.toml
./nnm/__init__.py
./nnm/version.py
./nnm/checkpoint/__init__.py
./nnm/checkpoint/ema.py
./nnm/checkpoint/saver.py
./nnm/dataset/__init__.py
./nnm/dataset/mnist.py
./nnm/dataset/swiss_roll.py
./nnm/diffusion/swiss_roll_ddpm.py
./nnm/gan/gan.py
./nnm/layers/__init__.py
./nnm/layers/drop.py
./nnm/layers/vit.py
./nnm/models/__init__.py
./nnm/models/davit.py
./nnm/vae/mnist_vae.py
./tests/test_vit.py
nnm/__init__.py
nnm/version.py
nnm.egg-info/PKG-INFO
nnm.egg-info/SOURCES.txt
nnm.egg-info/dependency_links.txt
nnm.egg-info/requires.txt
nnm.egg-info/top_level.txt
nnm/checkpoint/__init__.py
nnm/checkpoint/ema.py
nnm/checkpoint/saver.py
nnm/dataset/__init__.py
nnm/dataset/mnist.py
nnm/dataset/swiss_roll.py
nnm/diffusion/swiss_roll_ddpm.py
nnm/gan/gan.py
nnm/layers/__init__.py
nnm/layers/drop.py
nnm/layers/vit.py
nnm/models/__init__.py
nnm/models/davit.py
nnm/vae/mnist_vae.py
tests/test_vit.py