LICENSE
README.md
pyproject.toml
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/attention.py
nnm/layers/drop.py
nnm/layers/positional.py
nnm/layers/vit.py
nnm/metrics/__init__.py
nnm/metrics/belu.py
nnm/models/__init__.py
nnm/models/davit.py
nnm/models/fastvit.py
nnm/models/florence.py
nnm/utils/__init__.py
nnm/utils/grad.py
nnm/utils/param.py
nnm/vae/mnist_vae.py
tests/test_attention.py
tests/test_saver.py
tests/test_vit.py