LICENSE
README.md
pyproject.toml
src/neuraltoolkit/CLI.py
src/neuraltoolkit/__init__.py
src/neuraltoolkit/graph.py
src/neuraltoolkit/core/__init__.py
src/neuraltoolkit/core/device.py
src/neuraltoolkit/core/dtype.py
src/neuraltoolkit/core/no_grad.py
src/neuraltoolkit/core/parameter.py
src/neuraltoolkit/core/tensor.py
src/neuraltoolkit/core/tensor_ops.py
src/neuraltoolkit/data/__init__.py
src/neuraltoolkit/data/dataloader.py
src/neuraltoolkit/data/dataset.py
src/neuraltoolkit/data/subset.py
src/neuraltoolkit/datasets/__init__.py
src/neuraltoolkit/datasets/management/__init__.py
src/neuraltoolkit/datasets/management/cache.py
src/neuraltoolkit/datasets/management/data_resource.py
src/neuraltoolkit/datasets/management/downloader.py
src/neuraltoolkit/datasets/management/paths.py
src/neuraltoolkit/datasets/management/retrieve.py
src/neuraltoolkit/datasets/management/verify.py
src/neuraltoolkit/datasets/mnist/__init__.py
src/neuraltoolkit/datasets/mnist/mnist.py
src/neuraltoolkit/initializers/__init__.py
src/neuraltoolkit/initializers/glorot_initializer.py
src/neuraltoolkit/initializers/he_initializer.py
src/neuraltoolkit/loss/BCE.py
src/neuraltoolkit/loss/CCE.py
src/neuraltoolkit/loss/MSE.py
src/neuraltoolkit/loss/__init__.py
src/neuraltoolkit/modules/Registry.py
src/neuraltoolkit/modules/__init__.py
src/neuraltoolkit/modules/module.py
src/neuraltoolkit/modules/activations/__init__.py
src/neuraltoolkit/modules/activations/activation.py
src/neuraltoolkit/modules/activations/leakyrelu.py
src/neuraltoolkit/modules/activations/relu.py
src/neuraltoolkit/modules/activations/sigmoid.py
src/neuraltoolkit/modules/activations/softmax.py
src/neuraltoolkit/modules/activations/tanh.py
src/neuraltoolkit/modules/layers/__init__.py
src/neuraltoolkit/modules/layers/adaptive_max_pool2d.py
src/neuraltoolkit/modules/layers/conv2d.py
src/neuraltoolkit/modules/layers/dense.py
src/neuraltoolkit/modules/layers/flatten.py
src/neuraltoolkit/modules/layers/max_pool2d.py
src/neuraltoolkit/modules/models/__init__.py
src/neuraltoolkit/modules/models/sequential.py
src/neuraltoolkit/ops/__init__.py
src/neuraltoolkit/ops/data.py
src/neuraltoolkit/ops/image_processing.py
src/neuraltoolkit/optimizers/__init__.py
src/neuraltoolkit/optimizers/adagrad.py
src/neuraltoolkit/optimizers/adam.py
src/neuraltoolkit/optimizers/optimizer.py
src/neuraltoolkit/optimizers/rmsprop.py
src/neuraltoolkit/optimizers/sgd.py
src/neuraltoolkit/training/__init__.py
src/neuraltoolkit/training/config.py
src/neuraltoolkit/training/history.py
src/neuraltoolkit/training/trainer.py
src/ntk_ml.egg-info/PKG-INFO
src/ntk_ml.egg-info/SOURCES.txt
src/ntk_ml.egg-info/dependency_links.txt
src/ntk_ml.egg-info/requires.txt
src/ntk_ml.egg-info/top_level.txt
tests/test.py