LICENSE
README.md
pyproject.toml
src/torchsonn/__init__.py
src/torchsonn/layer.py
src/torchsonn/logger.py
src/torchsonn/model.py
src/torchsonn/plot_model.py
src/torchsonn/trainer.py
src/torchsonn/types.py
src/torchsonn/utils.py
src/torchsonn.egg-info/PKG-INFO
src/torchsonn.egg-info/SOURCES.txt
src/torchsonn.egg-info/dependency_links.txt
src/torchsonn.egg-info/requires.txt
src/torchsonn.egg-info/top_level.txt
src/torchsonn/config/__init__.py
src/torchsonn/config/schemas.py
src/torchsonn/data/__init__.py
src/torchsonn/data/dataset.py
src/torchsonn/data/preprocessing.py
src/torchsonn/loss/__init__.py
src/torchsonn/loss/errors.py
src/torchsonn/loss/norm_mse.py
src/torchsonn/modules/__init__.py
src/torchsonn/modules/base.py
src/torchsonn/modules/soft_binner.py
src/torchsonn/neurons/__init__.py
src/torchsonn/neurons/base.py
src/torchsonn/neurons/binary.py
src/torchsonn/neurons/poly.py
src/torchsonn/optimizers/__init__.py
src/torchsonn/optimizers/adam.py
src/torchsonn/optimizers/base.py
src/torchsonn/optimizers/lbfgs.py
src/torchsonn/optimizers/newton.py
src/torchsonn/optimizers/newton_lm.py
src/torchsonn/optimizers/sgd.py
src/torchsonn/schedulers/__init__.py
src/torchsonn/schedulers/base.py
src/torchsonn/schedulers/warmup_flat.py
tests/test_config.py
tests/test_data.py
tests/test_layer.py
tests/test_logger.py
tests/test_loss.py
tests/test_model.py
tests/test_modules.py
tests/test_neurons.py
tests/test_optimizers.py
tests/test_plot_model.py
tests/test_schedulers.py
tests/test_trainer.py
tests/test_trainer_smoke.py
tests/test_types.py
tests/test_utils.py