├── README.md ├── data ├── lotka-volterra │ ├── obs_stats.pkl │ ├── pilot_run_results.pkl │ ├── prior-observations-gaussian.npy │ ├── prior-observations.npy │ ├── prior-parameters-gaussian.npy │ └── prior-parameters.npy ├── mg1 │ ├── observed_data.pkl │ └── pilot_run_results.pkl └── nonlinear-gaussian │ └── true-posterior-samples.npy └── src ├── inference ├── __init__.py ├── apt.py ├── snl.py └── sre.py ├── mcmc ├── __init__.py ├── base.py ├── slice.py └── slice_numpy.py ├── nn_ ├── __init__.py ├── nde │ ├── __init__.py │ ├── made.py │ └── mdn.py └── nets │ ├── __init__.py │ ├── mlp.py │ └── resnet.py ├── nsf ├── __init__.py ├── distributions │ ├── __init__.py │ ├── base.py │ ├── discrete.py │ ├── discrete_test.py │ ├── mixture.py │ ├── normal.py │ ├── normal_test.py │ └── uniform.py ├── flows │ ├── __init__.py │ ├── autoregressive.py │ ├── autoregressive_test.py │ ├── base.py │ ├── base_test.py │ ├── realnvp.py │ └── realnvp_test.py └── transforms │ ├── __init__.py │ ├── autoregressive.py │ ├── autoregressive_test.py │ ├── base.py │ ├── base_test.py │ ├── conv.py │ ├── conv_test.py │ ├── coupling.py │ ├── coupling_test.py │ ├── linear.py │ ├── linear_test.py │ ├── lu.py │ ├── lu_test.py │ ├── made.py │ ├── made_test.py │ ├── nonlinearities.py │ ├── nonlinearities_test.py │ ├── normalization.py │ ├── normalization_test.py │ ├── orthogonal.py │ ├── orthogonal_test.py │ ├── permutations.py │ ├── permutations_test.py │ ├── qr.py │ ├── qr_test.py │ ├── reshape.py │ ├── reshape_test.py │ ├── splines │ ├── __init__.py │ ├── cubic.py │ ├── cubic_test.py │ ├── linear.py │ ├── linear_test.py │ ├── quadratic.py │ ├── quadratic_test.py │ ├── rational_quadratic.py │ └── rational_quadratic_test.py │ ├── standard.py │ ├── standard_test.py │ ├── svd.py │ ├── svd_test.py │ └── transform_test.py ├── simulators ├── __init__.py ├── linear_gaussian.py ├── lotka_volterra.py ├── lotka_volterra_pytorch.py ├── markov_jump_process.py ├── mg1.py ├── mg1_pytorch.py ├── nonlinear_gaussian.py ├── simulator.py ├── simutils.py └── two_moons.py └── utils ├── __init__.py ├── get_models.py ├── io.py ├── mmd.py ├── plot.py ├── torchutils.py ├── torchutils_test.py └── typechecks.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/README.md -------------------------------------------------------------------------------- /data/lotka-volterra/obs_stats.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/data/lotka-volterra/obs_stats.pkl -------------------------------------------------------------------------------- /data/lotka-volterra/pilot_run_results.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/data/lotka-volterra/pilot_run_results.pkl -------------------------------------------------------------------------------- /data/lotka-volterra/prior-observations-gaussian.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/data/lotka-volterra/prior-observations-gaussian.npy -------------------------------------------------------------------------------- /data/lotka-volterra/prior-observations.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/data/lotka-volterra/prior-observations.npy -------------------------------------------------------------------------------- /data/lotka-volterra/prior-parameters-gaussian.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/data/lotka-volterra/prior-parameters-gaussian.npy -------------------------------------------------------------------------------- /data/lotka-volterra/prior-parameters.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/data/lotka-volterra/prior-parameters.npy -------------------------------------------------------------------------------- /data/mg1/observed_data.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/data/mg1/observed_data.pkl -------------------------------------------------------------------------------- /data/mg1/pilot_run_results.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/data/mg1/pilot_run_results.pkl -------------------------------------------------------------------------------- /data/nonlinear-gaussian/true-posterior-samples.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/data/nonlinear-gaussian/true-posterior-samples.npy -------------------------------------------------------------------------------- /src/inference/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/inference/__init__.py -------------------------------------------------------------------------------- /src/inference/apt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/inference/apt.py -------------------------------------------------------------------------------- /src/inference/snl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/inference/snl.py -------------------------------------------------------------------------------- /src/inference/sre.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/inference/sre.py -------------------------------------------------------------------------------- /src/mcmc/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/mcmc/__init__.py -------------------------------------------------------------------------------- /src/mcmc/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/mcmc/base.py -------------------------------------------------------------------------------- /src/mcmc/slice.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/mcmc/slice.py -------------------------------------------------------------------------------- /src/mcmc/slice_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/mcmc/slice_numpy.py -------------------------------------------------------------------------------- /src/nn_/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/nn_/nde/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nn_/nde/__init__.py -------------------------------------------------------------------------------- /src/nn_/nde/made.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nn_/nde/made.py -------------------------------------------------------------------------------- /src/nn_/nde/mdn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nn_/nde/mdn.py -------------------------------------------------------------------------------- /src/nn_/nets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nn_/nets/__init__.py -------------------------------------------------------------------------------- /src/nn_/nets/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nn_/nets/mlp.py -------------------------------------------------------------------------------- /src/nn_/nets/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nn_/nets/resnet.py -------------------------------------------------------------------------------- /src/nsf/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/nsf/distributions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/distributions/__init__.py -------------------------------------------------------------------------------- /src/nsf/distributions/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/distributions/base.py -------------------------------------------------------------------------------- /src/nsf/distributions/discrete.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/distributions/discrete.py -------------------------------------------------------------------------------- /src/nsf/distributions/discrete_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/distributions/discrete_test.py -------------------------------------------------------------------------------- /src/nsf/distributions/mixture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/distributions/mixture.py -------------------------------------------------------------------------------- /src/nsf/distributions/normal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/distributions/normal.py -------------------------------------------------------------------------------- /src/nsf/distributions/normal_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/distributions/normal_test.py -------------------------------------------------------------------------------- /src/nsf/distributions/uniform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/distributions/uniform.py -------------------------------------------------------------------------------- /src/nsf/flows/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/flows/__init__.py -------------------------------------------------------------------------------- /src/nsf/flows/autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/flows/autoregressive.py -------------------------------------------------------------------------------- /src/nsf/flows/autoregressive_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/flows/autoregressive_test.py -------------------------------------------------------------------------------- /src/nsf/flows/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/flows/base.py -------------------------------------------------------------------------------- /src/nsf/flows/base_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/flows/base_test.py -------------------------------------------------------------------------------- /src/nsf/flows/realnvp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/flows/realnvp.py -------------------------------------------------------------------------------- /src/nsf/flows/realnvp_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/flows/realnvp_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/__init__.py -------------------------------------------------------------------------------- /src/nsf/transforms/autoregressive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/autoregressive.py -------------------------------------------------------------------------------- /src/nsf/transforms/autoregressive_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/autoregressive_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/base.py -------------------------------------------------------------------------------- /src/nsf/transforms/base_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/base_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/conv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/conv.py -------------------------------------------------------------------------------- /src/nsf/transforms/conv_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/conv_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/coupling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/coupling.py -------------------------------------------------------------------------------- /src/nsf/transforms/coupling_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/coupling_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/linear.py -------------------------------------------------------------------------------- /src/nsf/transforms/linear_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/linear_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/lu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/lu.py -------------------------------------------------------------------------------- /src/nsf/transforms/lu_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/lu_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/made.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/made.py -------------------------------------------------------------------------------- /src/nsf/transforms/made_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/made_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/nonlinearities.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/nonlinearities.py -------------------------------------------------------------------------------- /src/nsf/transforms/nonlinearities_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/nonlinearities_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/normalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/normalization.py -------------------------------------------------------------------------------- /src/nsf/transforms/normalization_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/normalization_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/orthogonal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/orthogonal.py -------------------------------------------------------------------------------- /src/nsf/transforms/orthogonal_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/orthogonal_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/permutations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/permutations.py -------------------------------------------------------------------------------- /src/nsf/transforms/permutations_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/permutations_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/qr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/qr.py -------------------------------------------------------------------------------- /src/nsf/transforms/qr_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/qr_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/reshape.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/reshape.py -------------------------------------------------------------------------------- /src/nsf/transforms/reshape_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/reshape_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/splines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/splines/__init__.py -------------------------------------------------------------------------------- /src/nsf/transforms/splines/cubic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/splines/cubic.py -------------------------------------------------------------------------------- /src/nsf/transforms/splines/cubic_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/splines/cubic_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/splines/linear.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/splines/linear.py -------------------------------------------------------------------------------- /src/nsf/transforms/splines/linear_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/splines/linear_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/splines/quadratic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/splines/quadratic.py -------------------------------------------------------------------------------- /src/nsf/transforms/splines/quadratic_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/splines/quadratic_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/splines/rational_quadratic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/splines/rational_quadratic.py -------------------------------------------------------------------------------- /src/nsf/transforms/splines/rational_quadratic_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/splines/rational_quadratic_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/standard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/standard.py -------------------------------------------------------------------------------- /src/nsf/transforms/standard_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/standard_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/svd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/svd.py -------------------------------------------------------------------------------- /src/nsf/transforms/svd_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/svd_test.py -------------------------------------------------------------------------------- /src/nsf/transforms/transform_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/nsf/transforms/transform_test.py -------------------------------------------------------------------------------- /src/simulators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/simulators/__init__.py -------------------------------------------------------------------------------- /src/simulators/linear_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/simulators/linear_gaussian.py -------------------------------------------------------------------------------- /src/simulators/lotka_volterra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/simulators/lotka_volterra.py -------------------------------------------------------------------------------- /src/simulators/lotka_volterra_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/simulators/lotka_volterra_pytorch.py -------------------------------------------------------------------------------- /src/simulators/markov_jump_process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/simulators/markov_jump_process.py -------------------------------------------------------------------------------- /src/simulators/mg1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/simulators/mg1.py -------------------------------------------------------------------------------- /src/simulators/mg1_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/simulators/mg1_pytorch.py -------------------------------------------------------------------------------- /src/simulators/nonlinear_gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/simulators/nonlinear_gaussian.py -------------------------------------------------------------------------------- /src/simulators/simulator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/simulators/simulator.py -------------------------------------------------------------------------------- /src/simulators/simutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/simulators/simutils.py -------------------------------------------------------------------------------- /src/simulators/two_moons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/simulators/two_moons.py -------------------------------------------------------------------------------- /src/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/utils/__init__.py -------------------------------------------------------------------------------- /src/utils/get_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/utils/get_models.py -------------------------------------------------------------------------------- /src/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/utils/io.py -------------------------------------------------------------------------------- /src/utils/mmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/utils/mmd.py -------------------------------------------------------------------------------- /src/utils/plot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/utils/plot.py -------------------------------------------------------------------------------- /src/utils/torchutils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/utils/torchutils.py -------------------------------------------------------------------------------- /src/utils/torchutils_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/utils/torchutils_test.py -------------------------------------------------------------------------------- /src/utils/typechecks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/conormdurkan/lfi/HEAD/src/utils/typechecks.py --------------------------------------------------------------------------------