Path Lines of Code MANIFEST.in 1 pyproject.toml 28 scripts/convert_ipynb_to_mdx.py 331 setup.cfg 59 setup.py 140 sphinx/source/conf.py 22 sphinx/source/docs.py 150 src/beanmachine/__init__.py 1 src/beanmachine/applications/__init__.py 1 src/beanmachine/applications/clara/__init__.py 1 src/beanmachine/applications/clara/nmc.py 170 src/beanmachine/applications/clara/nmc_df.py 148 src/beanmachine/applications/hme/__init__.py 18 src/beanmachine/applications/hme/abstract_linear_model.py 184 src/beanmachine/applications/hme/abstract_model.py 453 src/beanmachine/applications/hme/configs.py 59 src/beanmachine/applications/hme/interface.py 16 src/beanmachine/applications/hme/null_mixture_model.py 191 src/beanmachine/applications/hme/patsy_mixed.py 49 src/beanmachine/applications/hme/priors.py 99 src/beanmachine/graph/cavi.cpp 132 src/beanmachine/graph/distribution/bernoulli.cpp 110 src/beanmachine/graph/distribution/bernoulli.h 43 src/beanmachine/graph/distribution/bernoulli_logit.cpp 118 src/beanmachine/graph/distribution/bernoulli_logit.h 43 src/beanmachine/graph/distribution/bernoulli_noisy_or.cpp 119 src/beanmachine/graph/distribution/bernoulli_noisy_or.h 42 src/beanmachine/graph/distribution/beta.cpp 202 src/beanmachine/graph/distribution/beta.h 51 src/beanmachine/graph/distribution/bimixture.cpp 208 src/beanmachine/graph/distribution/bimixture.h 47 src/beanmachine/graph/distribution/binomial.cpp 134 src/beanmachine/graph/distribution/binomial.h 42 src/beanmachine/graph/distribution/categorical.cpp 96 src/beanmachine/graph/distribution/categorical.h 43 src/beanmachine/graph/distribution/dirichlet.cpp 81 src/beanmachine/graph/distribution/dirichlet.h 32 src/beanmachine/graph/distribution/distribution.cpp 184 src/beanmachine/graph/distribution/distribution.h 77 src/beanmachine/graph/distribution/flat.cpp 77 src/beanmachine/graph/distribution/flat.h 28 src/beanmachine/graph/distribution/gamma.cpp 172 src/beanmachine/graph/distribution/gamma.h 47 src/beanmachine/graph/distribution/half_cauchy.cpp 161 src/beanmachine/graph/distribution/half_cauchy.h 44 src/beanmachine/graph/distribution/half_normal.cpp 154 src/beanmachine/graph/distribution/half_normal.h 43 src/beanmachine/graph/distribution/normal.cpp 188 src/beanmachine/graph/distribution/normal.h 44 src/beanmachine/graph/distribution/student_t.cpp 262 src/beanmachine/graph/distribution/student_t.h 52 src/beanmachine/graph/distribution/tabular.cpp 97 src/beanmachine/graph/distribution/tabular.h 25 src/beanmachine/graph/factor/exp_product.cpp 70 src/beanmachine/graph/factor/exp_product.h 17 src/beanmachine/graph/factor/factor.cpp 22 src/beanmachine/graph/factor/factor.h 22 src/beanmachine/graph/gibbs.cpp 127 src/beanmachine/graph/global/global_mh.cpp 44 src/beanmachine/graph/global/global_mh.h 23 src/beanmachine/graph/global/global_state.cpp 186 src/beanmachine/graph/global/global_state.h 31 src/beanmachine/graph/global/hmc.cpp 14 src/beanmachine/graph/global/hmc.h 13 src/beanmachine/graph/global/nuts.cpp 13 src/beanmachine/graph/global/nuts.h 13 src/beanmachine/graph/global/proposer/global_proposer.h 20 src/beanmachine/graph/global/proposer/hmc_proposer.cpp 80 src/beanmachine/graph/global/proposer/hmc_proposer.h 25 src/beanmachine/graph/global/proposer/hmc_util.cpp 32 src/beanmachine/graph/global/proposer/hmc_util.h 21 src/beanmachine/graph/global/proposer/nuts_proposer.cpp 283 src/beanmachine/graph/global/proposer/nuts_proposer.h 72 src/beanmachine/graph/global/proposer/random_walk_proposer.cpp 21 src/beanmachine/graph/global/proposer/random_walk_proposer.h 12 src/beanmachine/graph/global/random_walk.cpp 10 src/beanmachine/graph/global/random_walk.h 10 src/beanmachine/graph/global/util.cpp 25 src/beanmachine/graph/global/util.h 6 src/beanmachine/graph/graph.cpp 1053 src/beanmachine/graph/graph.h 595 src/beanmachine/graph/graph.pyi 368 src/beanmachine/graph/mh.cpp 229 src/beanmachine/graph/mh.h 77 src/beanmachine/graph/nmc.cpp 39 src/beanmachine/graph/nmc.h 23 src/beanmachine/graph/nmc_stepper.h 25 src/beanmachine/graph/operator/backward.cpp 203 src/beanmachine/graph/operator/controlop.cpp 60 src/beanmachine/graph/operator/controlop.h 35 src/beanmachine/graph/operator/gradient.cpp 248 src/beanmachine/graph/operator/linalgop.cpp 244 src/beanmachine/graph/operator/linalgop.h 80 src/beanmachine/graph/operator/multiaryop.cpp 211 src/beanmachine/graph/operator/multiaryop.h 103 src/beanmachine/graph/operator/operator.cpp 58 src/beanmachine/graph/operator/operator.h 41 src/beanmachine/graph/operator/register.cpp 94 src/beanmachine/graph/operator/stochasticop.cpp 144 src/beanmachine/graph/operator/stochasticop.h 64 src/beanmachine/graph/operator/unaryop.cpp 495 src/beanmachine/graph/operator/unaryop.h 265 src/beanmachine/graph/perf_report.cpp 226 src/beanmachine/graph/profiler.cpp 23 src/beanmachine/graph/profiler.h 38 src/beanmachine/graph/proposer/beta.cpp 17 src/beanmachine/graph/proposer/beta.h 15 src/beanmachine/graph/proposer/default_initializer.cpp 33 src/beanmachine/graph/proposer/default_initializer.h 7 src/beanmachine/graph/proposer/delta.h 18 src/beanmachine/graph/proposer/from_probability_to_dirichlet_proposer_adapter.cpp 27 src/beanmachine/graph/proposer/from_probability_to_dirichlet_proposer_adapter.h 17 src/beanmachine/graph/proposer/gamma.cpp 16 src/beanmachine/graph/proposer/gamma.h 15 src/beanmachine/graph/proposer/mixture.cpp 33 src/beanmachine/graph/proposer/mixture.h 25 src/beanmachine/graph/proposer/nmc_proposer.cpp 81 src/beanmachine/graph/proposer/normal.cpp 16 src/beanmachine/graph/proposer/normal.h 15 src/beanmachine/graph/proposer/proposer.h 15 src/beanmachine/graph/proposer/trunc_cauchy.cpp 22 src/beanmachine/graph/proposer/trunc_cauchy.h 17 src/beanmachine/graph/pybindings.cpp 367 src/beanmachine/graph/pybindings.h 61 src/beanmachine/graph/rejection.cpp 41 src/beanmachine/graph/stepper/single_site/default_single_site_stepping_method.cpp 39 src/beanmachine/graph/stepper/single_site/default_single_site_stepping_method.h 20 src/beanmachine/graph/stepper/single_site/nmc_dirichlet_beta_single_site_stepping_method.cpp 62 src/beanmachine/graph/stepper/single_site/nmc_dirichlet_beta_single_site_stepping_method.h 19 src/beanmachine/graph/stepper/single_site/nmc_dirichlet_gamma_single_site_stepping_method.cpp 113 src/beanmachine/graph/stepper/single_site/nmc_dirichlet_gamma_single_site_stepping_method.h 27 src/beanmachine/graph/stepper/single_site/nmc_scalar_single_site_stepping_method.cpp 40 src/beanmachine/graph/stepper/single_site/nmc_scalar_single_site_stepping_method.h 19 src/beanmachine/graph/stepper/single_site/sequential_single_site_stepper.cpp 64 src/beanmachine/graph/stepper/single_site/sequential_single_site_stepper.h 26 src/beanmachine/graph/stepper/single_site/single_site_stepper.h 24 src/beanmachine/graph/stepper/single_site/single_site_stepping_method.h 16 src/beanmachine/graph/stepper/stepper.cpp 1 src/beanmachine/graph/stepper/stepper.h 15 src/beanmachine/graph/support.cpp 92 src/beanmachine/graph/to_dot.cpp 228 src/beanmachine/graph/transform/logtransform.cpp 60 src/beanmachine/graph/transform/transform.h 24 src/beanmachine/graph/util.cpp 98 src/beanmachine/graph/util.h 38 src/beanmachine/ppl/__init__.py 53 src/beanmachine/ppl/compiler/__init__.py 1 src/beanmachine/ppl/compiler/ast_patterns.py 346 src/beanmachine/ppl/compiler/ast_tools.py 30 src/beanmachine/ppl/compiler/beanstalk_common.py 2 src/beanmachine/ppl/compiler/bm_graph_builder.py 705 src/beanmachine/ppl/compiler/bm_to_bmg.py 411 src/beanmachine/ppl/compiler/bmg_node_types.py 103 src/beanmachine/ppl/compiler/bmg_nodes.py 611 src/beanmachine/ppl/compiler/bmg_requirements.py 193 src/beanmachine/ppl/compiler/bmg_types.py 391 src/beanmachine/ppl/compiler/error_report.py 85 src/beanmachine/ppl/compiler/fix_additions.py 32 src/beanmachine/ppl/compiler/fix_beta_conjugate_prior.py 119 src/beanmachine/ppl/compiler/fix_bool_arithmetic.py 53 src/beanmachine/ppl/compiler/fix_bool_comparisons.py 97 src/beanmachine/ppl/compiler/fix_logsumexp.py 24 src/beanmachine/ppl/compiler/fix_matrix_scale.py 19 src/beanmachine/ppl/compiler/fix_multiary_ops.py 64 src/beanmachine/ppl/compiler/fix_normal_conjugate_prior.py 86 src/beanmachine/ppl/compiler/fix_observations.py 38 src/beanmachine/ppl/compiler/fix_observe_true.py 53 src/beanmachine/ppl/compiler/fix_problem.py 52 src/beanmachine/ppl/compiler/fix_problems.py 69 src/beanmachine/ppl/compiler/fix_requirements.py 209 src/beanmachine/ppl/compiler/fix_unsupported.py 199 src/beanmachine/ppl/compiler/fix_vectorized_models.py 177 src/beanmachine/ppl/compiler/gen_bmg_cpp.py 168 src/beanmachine/ppl/compiler/gen_bmg_graph.py 105 src/beanmachine/ppl/compiler/gen_bmg_python.py 152 src/beanmachine/ppl/compiler/gen_builder.py 47 src/beanmachine/ppl/compiler/gen_dot.py 58 src/beanmachine/ppl/compiler/graph_labels.py 220 src/beanmachine/ppl/compiler/hint.py 6 src/beanmachine/ppl/compiler/internal_error.py 51 src/beanmachine/ppl/compiler/lattice_typer.py 256 src/beanmachine/ppl/compiler/patterns.py 375 src/beanmachine/ppl/compiler/performance_report.py 40 src/beanmachine/ppl/compiler/profiler.py 114 src/beanmachine/ppl/compiler/rules.py 679 src/beanmachine/ppl/compiler/runtime.py 990 src/beanmachine/ppl/compiler/single_assignment.py 1528 src/beanmachine/ppl/compiler/sizer.py 152 src/beanmachine/ppl/compiler/support.py 158 src/beanmachine/ppl/compiler/testlib/__init__.py 1 src/beanmachine/ppl/compiler/testlib/conjugate_models.py 37 src/beanmachine/ppl/compiler/typer_base.py 64 src/beanmachine/ppl/conftest.py 5 src/beanmachine/ppl/diagnostics/__init__.py 2 src/beanmachine/ppl/diagnostics/common_plots.py 70 src/beanmachine/ppl/diagnostics/common_statistics.py 87 src/beanmachine/ppl/diagnostics/diagnostics.py 208 src/beanmachine/ppl/distributions/__init__.py 3 src/beanmachine/ppl/distributions/flat.py 15 src/beanmachine/ppl/distributions/unit.py 38 src/beanmachine/ppl/experimental/__init__.py 1 src/beanmachine/ppl/experimental/abc/__init__.py 1 src/beanmachine/ppl/experimental/abc/abc_infer.py 57 src/beanmachine/ppl/experimental/abc/abc_smc_infer.py 140 src/beanmachine/ppl/experimental/abc/adaptive_abc_smc_infer.py 183 src/beanmachine/ppl/experimental/gp/__init__.py 3 src/beanmachine/ppl/experimental/gp/kernels.py 157 src/beanmachine/ppl/experimental/gp/likelihoods.py 80 src/beanmachine/ppl/experimental/gp/models.py 65 src/beanmachine/ppl/experimental/inference_compilation/__init__.py 1 src/beanmachine/ppl/experimental/inference_compilation/ic_infer.py 399 src/beanmachine/ppl/experimental/inference_compilation/utils.py 12 src/beanmachine/ppl/experimental/neutra/__init__.py 1 src/beanmachine/ppl/experimental/neutra/iafflow.py 63 src/beanmachine/ppl/experimental/neutra/iaflayer.py 55 src/beanmachine/ppl/experimental/neutra/iafmcmc_infer.py 54 src/beanmachine/ppl/experimental/neutra/iafmcmc_proposer.py 92 src/beanmachine/ppl/experimental/neutra/maskedautoencoder.py 76 src/beanmachine/ppl/experimental/neutra/maskedlinear.py 20 src/beanmachine/ppl/experimental/neutra/train.py 36 src/beanmachine/ppl/experimental/variable_elimination/__init__.py 1 src/beanmachine/ppl/experimental/variable_elimination/util.py 2 src/beanmachine/ppl/experimental/variable_elimination/variable_elimination.py 37 src/beanmachine/ppl/experimental/vi/__init__.py 1 src/beanmachine/ppl/experimental/vi/mean_field_variational_approximation.py 111 src/beanmachine/ppl/experimental/vi/optim.py 160 src/beanmachine/ppl/experimental/vi/variational_infer.py 205 src/beanmachine/ppl/inference/__init__.py 40 src/beanmachine/ppl/inference/base_inference.py 104 src/beanmachine/ppl/inference/bmg_inference.py 211 src/beanmachine/ppl/inference/compositional_infer.py 144 src/beanmachine/ppl/inference/hmc_inference.py 79 src/beanmachine/ppl/inference/monte_carlo_samples.py 150 src/beanmachine/ppl/inference/nuts_inference.py 91 src/beanmachine/ppl/inference/predictive.py 100 src/beanmachine/ppl/inference/proposer/__init__.py 1 src/beanmachine/ppl/inference/proposer/base_proposer.py 12 src/beanmachine/ppl/inference/proposer/base_single_site_proposer.py 37 src/beanmachine/ppl/inference/proposer/hmc_proposer.py 217 src/beanmachine/ppl/inference/proposer/hmc_utils.py 147 src/beanmachine/ppl/inference/proposer/nmc/__init__.py 14 src/beanmachine/ppl/inference/proposer/nmc/single_site_half_space_nmc_proposer.py 62 src/beanmachine/ppl/inference/proposer/nmc/single_site_real_space_nmc_proposer.py 151 src/beanmachine/ppl/inference/proposer/nmc/single_site_simplex_space_nmc_proposer.py 60 src/beanmachine/ppl/inference/proposer/nuts_proposer.py 232 src/beanmachine/ppl/inference/proposer/sequential_proposer.py 21 src/beanmachine/ppl/inference/proposer/single_site_ancestral_proposer.py 8 src/beanmachine/ppl/inference/proposer/single_site_random_walk_proposer.py 86 src/beanmachine/ppl/inference/proposer/single_site_uniform_proposer.py 15 src/beanmachine/ppl/inference/sampler.py 71 src/beanmachine/ppl/inference/single_site_ancestral_mh.py 9 src/beanmachine/ppl/inference/single_site_inference.py 29 src/beanmachine/ppl/inference/single_site_nmc.py 65 src/beanmachine/ppl/inference/single_site_random_walk.py 28 src/beanmachine/ppl/inference/single_site_uniform_mh.py 9 src/beanmachine/ppl/inference/utils.py 84 src/beanmachine/ppl/legacy/__init__.py 1 src/beanmachine/ppl/legacy/inference/__init__.py 33 src/beanmachine/ppl/legacy/inference/abstract_infer.py 152 src/beanmachine/ppl/legacy/inference/abstract_mh_infer.py 288 src/beanmachine/ppl/legacy/inference/compositional_infer.py 81 src/beanmachine/ppl/legacy/inference/proposer/__init__.py 18 src/beanmachine/ppl/legacy/inference/proposer/abstract_single_site_proposer.py 33 src/beanmachine/ppl/legacy/inference/proposer/abstract_single_site_single_step_proposer.py 81 src/beanmachine/ppl/legacy/inference/proposer/newtonian_monte_carlo_utils.py 81 src/beanmachine/ppl/legacy/inference/proposer/normal_eig.py 32 src/beanmachine/ppl/legacy/inference/proposer/single_site_ancestral_proposer.py 23 src/beanmachine/ppl/legacy/inference/proposer/single_site_half_space_newtonian_monte_carlo_proposer.py 75 src/beanmachine/ppl/legacy/inference/proposer/single_site_hamiltonian_monte_carlo_proposer.py 257 src/beanmachine/ppl/legacy/inference/proposer/single_site_newtonian_monte_carlo_proposer.py 106 src/beanmachine/ppl/legacy/inference/proposer/single_site_no_u_turn_sampler_proposer.py 428 src/beanmachine/ppl/legacy/inference/proposer/single_site_random_walk_proposer.py 150 src/beanmachine/ppl/legacy/inference/proposer/single_site_real_space_newtonian_monte_carlo_proposer.py 171 src/beanmachine/ppl/legacy/inference/proposer/single_site_simplex_newtonian_monte_carlo_proposer.py 75 src/beanmachine/ppl/legacy/inference/proposer/single_site_uniform_proposer.py 54 src/beanmachine/ppl/legacy/inference/rejection_sampling_infer.py 97 src/beanmachine/ppl/legacy/inference/sampler.py 58 src/beanmachine/ppl/legacy/inference/single_site_ancestral_mh.py 11 src/beanmachine/ppl/legacy/inference/single_site_hamiltonian_monte_carlo.py 29 src/beanmachine/ppl/legacy/inference/single_site_newtonian_monte_carlo.py 27 src/beanmachine/ppl/legacy/inference/single_site_no_u_turn_sampler.py 27 src/beanmachine/ppl/legacy/inference/single_site_random_walk.py 18 src/beanmachine/ppl/legacy/inference/single_site_uniform_mh.py 17 src/beanmachine/ppl/legacy/inference/utils.py 15 src/beanmachine/ppl/legacy/world/__init__.py 20 src/beanmachine/ppl/legacy/world/diff.py 40 src/beanmachine/ppl/legacy/world/diff_stack.py 90 src/beanmachine/ppl/legacy/world/marginal_table.py 32 src/beanmachine/ppl/legacy/world/variable.py 128 src/beanmachine/ppl/legacy/world/world.py 439 src/beanmachine/ppl/legacy/world/world_vars.py 41 src/beanmachine/ppl/model/__init__.py 19 src/beanmachine/ppl/model/rv_identifier.py 21 src/beanmachine/ppl/model/statistical_model.py 49 src/beanmachine/ppl/model/utils.py 28 src/beanmachine/ppl/testlib/__init__.py 1 src/beanmachine/ppl/testlib/abstract_conjugate.py 324 src/beanmachine/ppl/testlib/hypothesis_testing.py 81 src/beanmachine/ppl/utils/__init__.py 14 src/beanmachine/ppl/utils/dotbuilder.py 207 src/beanmachine/ppl/utils/equivalence.py 22 src/beanmachine/ppl/utils/graph.py 215 src/beanmachine/ppl/utils/item_counter.py 19 src/beanmachine/ppl/utils/memoize.py 86 src/beanmachine/ppl/utils/set_of_tensors.py 20 src/beanmachine/ppl/utils/tensorops.py 38 src/beanmachine/ppl/utils/treeprinter.py 67 src/beanmachine/ppl/utils/unique_name.py 21 src/beanmachine/ppl/world/__init__.py 21 src/beanmachine/ppl/world/base_world.py 21 src/beanmachine/ppl/world/initialize_fn.py 16 src/beanmachine/ppl/world/utils.py 94 src/beanmachine/ppl/world/variable.py 26 src/beanmachine/ppl/world/world.py 109 src/beanmachine/tutorials/__init__.py 1 src/beanmachine/tutorials/utils/__init__.py 1 src/beanmachine/tutorials/utils/baseball.py 615 src/beanmachine/tutorials/utils/etl.py 37 src/beanmachine/tutorials/utils/hearts.py 580 src/beanmachine/tutorials/utils/nba.py 374 src/beanmachine/tutorials/utils/plots.py 170 src/beanmachine/tutorials/utils/radon.py 519 website/babel.config.js 3 website/documentation.toml 40 website/docusaurus.config.js 111 website/sidebars.js 55 website/src/components/CellOutput.jsx 28 website/src/components/LinkButtons.jsx 12 website/src/components/Plotting.jsx 37 website/src/css/custom.css 16 website/src/pages/index.js 112 website/src/pages/styles.module.css 29