├── .gitignore ├── LICENSE ├── README.md ├── arguments ├── __init__.py └── default.py ├── compute_exact_free_energy.py ├── control_variate.py ├── decay.py ├── distributions.py ├── elbo.py ├── environment.yml ├── fit_ising_made.py ├── flow ├── __init__.py ├── binary_mixture.py ├── inverse_autoregressive.py ├── masked_autoregressive.py ├── real_nvp.py ├── sequential.py └── util.py ├── grid.py ├── hier_elbo.py ├── importance_sampling.py ├── jobs.py ├── log.py ├── main.py ├── mean_field_elbo.py ├── model ├── __init__.py ├── grid.py ├── ising.py ├── poisson.py └── sk.py ├── moving_averages.py ├── network ├── __init__.py ├── channel_made.py ├── conv.py ├── invertible_network.py ├── made.py ├── mask.py ├── masked_linear.py ├── scalar_linear.py └── scalar_residual.py ├── notebook ├── nearest-neighbor-convolution.nb ├── plot-annealing-schedule.ipynb ├── plot-energy-function.ipynb ├── plot-free-energy-vs-time-hvm-and-van.ipynb ├── plot-free-energy-vs-time-hvm-single-run.ipynb ├── plot-free-energy.ipynb ├── plot-relative-error-vs-time.ipynb ├── plot-toy-correlated-posterior.ipynb ├── pytorch-grad-matrix-vector-product.ipynb └── visualize-weight-masks.ipynb ├── print_exact_free_energy.py ├── proximity.py ├── stats.py ├── tests ├── data.py ├── fit_optimal_recursive.py ├── moons.py ├── test_bbvi.py ├── test_conv.py ├── test_inverse_autoregressive_flow.py ├── test_masked_autoregressive_flow.py ├── test_numpy_checkerboard.py ├── test_real_nvp.py └── test_toy_grid.py ├── util.py ├── variational ├── __init__.py ├── likelihood.py ├── posterior.py ├── prior.py └── variational_made.py └── visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/README.md -------------------------------------------------------------------------------- /arguments/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/arguments/__init__.py -------------------------------------------------------------------------------- /arguments/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/arguments/default.py -------------------------------------------------------------------------------- /compute_exact_free_energy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/compute_exact_free_energy.py -------------------------------------------------------------------------------- /control_variate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/control_variate.py -------------------------------------------------------------------------------- /decay.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/decay.py -------------------------------------------------------------------------------- /distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/distributions.py -------------------------------------------------------------------------------- /elbo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/elbo.py -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/environment.yml -------------------------------------------------------------------------------- /fit_ising_made.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/fit_ising_made.py -------------------------------------------------------------------------------- /flow/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/flow/__init__.py -------------------------------------------------------------------------------- /flow/binary_mixture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/flow/binary_mixture.py -------------------------------------------------------------------------------- /flow/inverse_autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/flow/inverse_autoregressive.py -------------------------------------------------------------------------------- /flow/masked_autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/flow/masked_autoregressive.py -------------------------------------------------------------------------------- /flow/real_nvp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/flow/real_nvp.py -------------------------------------------------------------------------------- /flow/sequential.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/flow/sequential.py -------------------------------------------------------------------------------- /flow/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/flow/util.py -------------------------------------------------------------------------------- /grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/grid.py -------------------------------------------------------------------------------- /hier_elbo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/hier_elbo.py -------------------------------------------------------------------------------- /importance_sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/importance_sampling.py -------------------------------------------------------------------------------- /jobs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/jobs.py -------------------------------------------------------------------------------- /log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/log.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/main.py -------------------------------------------------------------------------------- /mean_field_elbo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/mean_field_elbo.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/model/grid.py -------------------------------------------------------------------------------- /model/ising.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/model/ising.py -------------------------------------------------------------------------------- /model/poisson.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/model/poisson.py -------------------------------------------------------------------------------- /model/sk.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/model/sk.py -------------------------------------------------------------------------------- /moving_averages.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/moving_averages.py -------------------------------------------------------------------------------- /network/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/network/__init__.py -------------------------------------------------------------------------------- /network/channel_made.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/network/channel_made.py -------------------------------------------------------------------------------- /network/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/network/conv.py -------------------------------------------------------------------------------- /network/invertible_network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/network/invertible_network.py -------------------------------------------------------------------------------- /network/made.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/network/made.py -------------------------------------------------------------------------------- /network/mask.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/network/mask.py -------------------------------------------------------------------------------- /network/masked_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/network/masked_linear.py -------------------------------------------------------------------------------- /network/scalar_linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/network/scalar_linear.py -------------------------------------------------------------------------------- /network/scalar_residual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/network/scalar_residual.py -------------------------------------------------------------------------------- /notebook/nearest-neighbor-convolution.nb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/notebook/nearest-neighbor-convolution.nb -------------------------------------------------------------------------------- /notebook/plot-annealing-schedule.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/notebook/plot-annealing-schedule.ipynb -------------------------------------------------------------------------------- /notebook/plot-energy-function.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/notebook/plot-energy-function.ipynb -------------------------------------------------------------------------------- /notebook/plot-free-energy-vs-time-hvm-and-van.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/notebook/plot-free-energy-vs-time-hvm-and-van.ipynb -------------------------------------------------------------------------------- /notebook/plot-free-energy-vs-time-hvm-single-run.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/notebook/plot-free-energy-vs-time-hvm-single-run.ipynb -------------------------------------------------------------------------------- /notebook/plot-free-energy.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/notebook/plot-free-energy.ipynb -------------------------------------------------------------------------------- /notebook/plot-relative-error-vs-time.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/notebook/plot-relative-error-vs-time.ipynb -------------------------------------------------------------------------------- /notebook/plot-toy-correlated-posterior.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/notebook/plot-toy-correlated-posterior.ipynb -------------------------------------------------------------------------------- /notebook/pytorch-grad-matrix-vector-product.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/notebook/pytorch-grad-matrix-vector-product.ipynb -------------------------------------------------------------------------------- /notebook/visualize-weight-masks.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/notebook/visualize-weight-masks.ipynb -------------------------------------------------------------------------------- /print_exact_free_energy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/print_exact_free_energy.py -------------------------------------------------------------------------------- /proximity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/proximity.py -------------------------------------------------------------------------------- /stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/stats.py -------------------------------------------------------------------------------- /tests/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/tests/data.py -------------------------------------------------------------------------------- /tests/fit_optimal_recursive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/tests/fit_optimal_recursive.py -------------------------------------------------------------------------------- /tests/moons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/tests/moons.py -------------------------------------------------------------------------------- /tests/test_bbvi.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/tests/test_bbvi.py -------------------------------------------------------------------------------- /tests/test_conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/tests/test_conv.py -------------------------------------------------------------------------------- /tests/test_inverse_autoregressive_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/tests/test_inverse_autoregressive_flow.py -------------------------------------------------------------------------------- /tests/test_masked_autoregressive_flow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/tests/test_masked_autoregressive_flow.py -------------------------------------------------------------------------------- /tests/test_numpy_checkerboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/tests/test_numpy_checkerboard.py -------------------------------------------------------------------------------- /tests/test_real_nvp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/tests/test_real_nvp.py -------------------------------------------------------------------------------- /tests/test_toy_grid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/tests/test_toy_grid.py -------------------------------------------------------------------------------- /util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/util.py -------------------------------------------------------------------------------- /variational/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/variational/__init__.py -------------------------------------------------------------------------------- /variational/likelihood.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/variational/likelihood.py -------------------------------------------------------------------------------- /variational/posterior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/variational/posterior.py -------------------------------------------------------------------------------- /variational/prior.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/variational/prior.py -------------------------------------------------------------------------------- /variational/variational_made.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/variational/variational_made.py -------------------------------------------------------------------------------- /visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jaanli/hierarchical-variational-models-physics/HEAD/visualize.py --------------------------------------------------------------------------------