.gitignore
LICENSE
README.md
pyproject.toml
.github/workflows/release.yml
.github/workflows/tests.yml
.vscode/settings.json
assets/readme.png
nn_zoo/__init__.py
nn_zoo.egg-info/PKG-INFO
nn_zoo.egg-info/SOURCES.txt
nn_zoo.egg-info/dependency_links.txt
nn_zoo.egg-info/requires.txt
nn_zoo.egg-info/top_level.txt
nn_zoo/datamodules/__init__.py
nn_zoo/datamodules/cifar.py
nn_zoo/datamodules/ffhq.py
nn_zoo/datamodules/mnist.py
nn_zoo/models/__init__.py
nn_zoo/models/gpt.py
nn_zoo/models/unet.py
nn_zoo/models/vqvae.py
nn_zoo/models/components/__init__.py
nn_zoo/models/components/attention.py
nn_zoo/models/components/residual.py
nn_zoo/models/components/vq.py
nn_zoo/trainers/__init__.py
nn_zoo/trainers/diffusion.py
nn_zoo/trainers/seq_to_seq.py
nn_zoo/trainers/vqvae_trainer.py
tests/models/gpt_test.py
tests/models/unet_test.py
tests/models/vqvae_test.py
tests/models/components/attention_test.py
tests/models/components/residual_test.py
tests/trainers/vqvae_trainer_test.py