README.md
pyproject.toml
setup.py
axon/__init__.py
axon/experiment.py
axon/functions.py
axon/libaxon.so
axon/axon_bindings/__init__.py
axon/axon_bindings/c_function_signatures.py
axon/axon_bindings/c_library_loader.py
axon/axon_bindings/c_wrapper_functions.py
axon/axon_bindings/ctypes_definitions.py
axon/axon_bindings/utils.py
axon/core/__init__.py
axon/core/__main__.py
axon/core/buffer.py
axon/core/tensor.py
axon/data/__init__.py
axon/data/dataloader.py
axon/data/dataset.py
axon/metrics/__init__.py
axon/metrics/bce.py
axon/metrics/mse.py
axon/metrics/nll_loss.py
axon/nn/__init__.py
axon/nn/activations.py
axon/nn/conv.py
axon/nn/init.py
axon/nn/linear.py
axon/nn/module.py
axon/nn/pipeline.py
axon/ops/__init__.py
axon/ops/bop.py
axon/ops/mop.py
axon/ops/op.py
axon/ops/rop.py
axon/ops/uop.py
axon/optim/__init__.py
axon/optim/adam.py
axon/optim/optimizer.py
axon/optim/sgd.py
axon/utils/model_io.py
axon_dl.egg-info/PKG-INFO
axon_dl.egg-info/SOURCES.txt
axon_dl.egg-info/dependency_links.txt
axon_dl.egg-info/not-zip-safe
axon_dl.egg-info/requires.txt
axon_dl.egg-info/top_level.txt
tests/test_core.py
tests/test_data.py
tests/test_experiment.py
tests/test_functions.py
tests/test_metrics.py
tests/test_model_io.py
tests/test_nn.py
tests/test_optim.py