.gitignore
LICENSE
README.md
dev-requirements.txt
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/image/__init__.py
nn_zoo/datamodules/image/cifar.py
nn_zoo/datamodules/image/ffhq.py
nn_zoo/datamodules/image/mnist.py
nn_zoo/models/__init__.py
nn_zoo/models/utils.py
nn_zoo/models/components/__init__.py
nn_zoo/models/components/attention.py
nn_zoo/models/components/depthwise.py
nn_zoo/models/components/residual.py
nn_zoo/models/components/vq.py
nn_zoo/trainers/__init__.py
nn_zoo/trainers/classifier.py
nn_zoo/trainers/diffusion.py
nn_zoo/trainers/encoder.py
nn_zoo/trainers/encodergan.py
nn_zoo/trainers/seq_to_seq.py
nn_zoo/trainers/utils.py
tests/models/utils_test.py
tests/models/components/attention_test.py
tests/models/components/depthwise_test.py
tests/models/components/residual_test.py
tests/models/components/vq_test.py