LICENSE.txt
README.md
pyproject.toml
src/bonsai_nn_library.egg-info/PKG-INFO
src/bonsai_nn_library.egg-info/SOURCES.txt
src/bonsai_nn_library.egg-info/dependency_links.txt
src/bonsai_nn_library.egg-info/requires.txt
src/bonsai_nn_library.egg-info/top_level.txt
src/nn_lib/__init__.py
src/nn_lib/analysis/__init__.py
src/nn_lib/analysis/pca.py
src/nn_lib/analysis/regression.py
src/nn_lib/analysis/ntk/__init__.py
src/nn_lib/analysis/ntk/learnability.py
src/nn_lib/analysis/ntk/linearize.py
src/nn_lib/analysis/ntk/ntk.py
src/nn_lib/analysis/similarity/__init__.py
src/nn_lib/analysis/similarity/cka.py
src/nn_lib/datasets/__init__.py
src/nn_lib/datasets/base.py
src/nn_lib/datasets/cifar.py
src/nn_lib/datasets/coco_semantic_segmentation.py
src/nn_lib/datasets/enums.py
src/nn_lib/datasets/imagenet.py
src/nn_lib/datasets/mnist.py
src/nn_lib/datasets/transforms.py
src/nn_lib/datasets/ucf101.py
src/nn_lib/datasets/vanhateren.py
src/nn_lib/jobs/__init__.py
src/nn_lib/jobs/dispatch.py
src/nn_lib/models/__init__.py
src/nn_lib/models/fancy_layers.py
src/nn_lib/models/graph_module_plus.py
src/nn_lib/models/graph_utils.py
src/nn_lib/models/parametrizations.py
src/nn_lib/models/sparse_auto_encoder.py
src/nn_lib/models/vector_quantization.py
src/nn_lib/optim/__init__.py
src/nn_lib/optim/lr_finder.py
src/nn_lib/utils/__init__.py
src/nn_lib/utils/cli.py
src/nn_lib/utils/generic.py
src/nn_lib/utils/mlflow.py
src/nn_lib/utils/models.py
src/nn_lib/utils/profile.py
tests/test_cli_utils.py
tests/test_datasets.py
tests/test_fancy_layers.py
tests/test_graph_module_plus.py
tests/test_lr_finder.py
tests/test_mlflow_utils.py
tests/test_ntk.py
tests/test_ntk_learnability.py
tests/test_ntk_linearization.py
tests/test_pca.py
tests/test_regression.py