├── .envrc ├── .github └── workflows │ └── python-app.yml ├── .gitignore ├── CITATION.cff ├── LICENSE.md ├── PROJECT_NAME ├── README.md ├── args.py ├── benchmark └── benchmark_simple_vs_og.py ├── conf ├── config.yaml └── experiment │ ├── cifar-10.yaml │ ├── cifar-100.yaml │ ├── mnist.yaml │ ├── svhn.yaml │ └── synth.yaml ├── exp_utils.py ├── main.py ├── main_pl.py ├── main_pwl.py ├── models_pl.py ├── notebooks ├── iris_classification.ipynb ├── mnist.ipynb └── multivariate_normal.ipynb ├── pyproject.toml ├── res ├── mnist-0-1-rec.png ├── mnist-0-1-samples.png ├── mnist_classification.png ├── mnist_samples.png └── mnist_train_val_test_acc.png ├── simple_einet ├── __init__.py ├── abstract_layers.py ├── conv_pc.py ├── data.py ├── dist.py ├── einet.py ├── histogram.py ├── layers │ ├── __init__.py │ ├── distributions │ │ ├── __init__.py │ │ ├── abstract_leaf.py │ │ ├── bernoulli.py │ │ ├── binomial.py │ │ ├── categorical.py │ │ ├── mixture.py │ │ ├── multidistribution.py │ │ ├── multivariate_normal.py │ │ ├── normal.py │ │ ├── piecewise_linear.py │ │ └── utils.py │ ├── einsum.py │ ├── factorized_leaf.py │ ├── linsum.py │ ├── mixing.py │ ├── product.py │ └── sum.py ├── mixture.py ├── sampling_utils.py ├── type_checks.py └── utils.py └── tests ├── __init__.py ├── layers ├── __init__.py ├── distributions │ └── __init__.py ├── test_einsum_layer.py ├── test_linsum_layer.py ├── test_mixing_layer.py ├── test_sum_layer.py └── test_utils.py └── test_einet.py /.envrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/.envrc -------------------------------------------------------------------------------- /.github/workflows/python-app.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/.github/workflows/python-app.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/.gitignore -------------------------------------------------------------------------------- /CITATION.cff: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/CITATION.cff -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/LICENSE.md -------------------------------------------------------------------------------- /PROJECT_NAME: -------------------------------------------------------------------------------- 1 | simple-einet 2 | -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/README.md -------------------------------------------------------------------------------- /args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/args.py -------------------------------------------------------------------------------- /benchmark/benchmark_simple_vs_og.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/benchmark/benchmark_simple_vs_og.py -------------------------------------------------------------------------------- /conf/config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/conf/config.yaml -------------------------------------------------------------------------------- /conf/experiment/cifar-10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/conf/experiment/cifar-10.yaml -------------------------------------------------------------------------------- /conf/experiment/cifar-100.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/conf/experiment/cifar-100.yaml -------------------------------------------------------------------------------- /conf/experiment/mnist.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/conf/experiment/mnist.yaml -------------------------------------------------------------------------------- /conf/experiment/svhn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/conf/experiment/svhn.yaml -------------------------------------------------------------------------------- /conf/experiment/synth.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/conf/experiment/synth.yaml -------------------------------------------------------------------------------- /exp_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/exp_utils.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/main.py -------------------------------------------------------------------------------- /main_pl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/main_pl.py -------------------------------------------------------------------------------- /main_pwl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/main_pwl.py -------------------------------------------------------------------------------- /models_pl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/models_pl.py -------------------------------------------------------------------------------- /notebooks/iris_classification.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/notebooks/iris_classification.ipynb -------------------------------------------------------------------------------- /notebooks/mnist.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/notebooks/mnist.ipynb -------------------------------------------------------------------------------- /notebooks/multivariate_normal.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/notebooks/multivariate_normal.ipynb -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/pyproject.toml -------------------------------------------------------------------------------- /res/mnist-0-1-rec.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/res/mnist-0-1-rec.png -------------------------------------------------------------------------------- /res/mnist-0-1-samples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/res/mnist-0-1-samples.png -------------------------------------------------------------------------------- /res/mnist_classification.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/res/mnist_classification.png -------------------------------------------------------------------------------- /res/mnist_samples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/res/mnist_samples.png -------------------------------------------------------------------------------- /res/mnist_train_val_test_acc.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/res/mnist_train_val_test_acc.png -------------------------------------------------------------------------------- /simple_einet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/__init__.py -------------------------------------------------------------------------------- /simple_einet/abstract_layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/abstract_layers.py -------------------------------------------------------------------------------- /simple_einet/conv_pc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/conv_pc.py -------------------------------------------------------------------------------- /simple_einet/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/data.py -------------------------------------------------------------------------------- /simple_einet/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/dist.py -------------------------------------------------------------------------------- /simple_einet/einet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/einet.py -------------------------------------------------------------------------------- /simple_einet/histogram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/histogram.py -------------------------------------------------------------------------------- /simple_einet/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /simple_einet/layers/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/distributions/__init__.py -------------------------------------------------------------------------------- /simple_einet/layers/distributions/abstract_leaf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/distributions/abstract_leaf.py -------------------------------------------------------------------------------- /simple_einet/layers/distributions/bernoulli.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/distributions/bernoulli.py -------------------------------------------------------------------------------- /simple_einet/layers/distributions/binomial.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/distributions/binomial.py -------------------------------------------------------------------------------- /simple_einet/layers/distributions/categorical.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/distributions/categorical.py -------------------------------------------------------------------------------- /simple_einet/layers/distributions/mixture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/distributions/mixture.py -------------------------------------------------------------------------------- /simple_einet/layers/distributions/multidistribution.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/distributions/multidistribution.py -------------------------------------------------------------------------------- /simple_einet/layers/distributions/multivariate_normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/distributions/multivariate_normal.py -------------------------------------------------------------------------------- /simple_einet/layers/distributions/normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/distributions/normal.py -------------------------------------------------------------------------------- /simple_einet/layers/distributions/piecewise_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/distributions/piecewise_linear.py -------------------------------------------------------------------------------- /simple_einet/layers/distributions/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/distributions/utils.py -------------------------------------------------------------------------------- /simple_einet/layers/einsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/einsum.py -------------------------------------------------------------------------------- /simple_einet/layers/factorized_leaf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/factorized_leaf.py -------------------------------------------------------------------------------- /simple_einet/layers/linsum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/linsum.py -------------------------------------------------------------------------------- /simple_einet/layers/mixing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/mixing.py -------------------------------------------------------------------------------- /simple_einet/layers/product.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/product.py -------------------------------------------------------------------------------- /simple_einet/layers/sum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/layers/sum.py -------------------------------------------------------------------------------- /simple_einet/mixture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/mixture.py -------------------------------------------------------------------------------- /simple_einet/sampling_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/sampling_utils.py -------------------------------------------------------------------------------- /simple_einet/type_checks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/type_checks.py -------------------------------------------------------------------------------- /simple_einet/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/simple_einet/utils.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/layers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/layers/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/layers/test_einsum_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/tests/layers/test_einsum_layer.py -------------------------------------------------------------------------------- /tests/layers/test_linsum_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/tests/layers/test_linsum_layer.py -------------------------------------------------------------------------------- /tests/layers/test_mixing_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/tests/layers/test_mixing_layer.py -------------------------------------------------------------------------------- /tests/layers/test_sum_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/tests/layers/test_sum_layer.py -------------------------------------------------------------------------------- /tests/layers/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/tests/layers/test_utils.py -------------------------------------------------------------------------------- /tests/test_einet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/braun-steven/simple-einet/HEAD/tests/test_einet.py --------------------------------------------------------------------------------