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/env.py
src/nn_lib/trainer.py
src/nn_lib/analysis/__init__.py
src/nn_lib/analysis/ntk.py
src/nn_lib/analysis/pca.py
src/nn_lib/analysis/regression.py
src/nn_lib/analysis/kernels/__init__.py
src/nn_lib/analysis/kernels/base.py
src/nn_lib/analysis/kernels/brownian.py
src/nn_lib/analysis/kernels/linear.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/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/lit_classifier.py
src/nn_lib/models/parametrizations.py
src/nn_lib/models/sparse_auto_encoder.py
src/nn_lib/models/utils.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
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_pca.py
tests/test_regression.py