.gitignore
.readthedocs.yaml
CODE_OF_CONDUCT.md
CONTRIBUTING.md
LICENSE
README.md
mypy.ini
pyproject.toml
ruff.toml
.github/PULL_REQUEST_TEMPLATE.md
.github/ISSUE_TEMPLATE/bug_report.yml
.github/ISSUE_TEMPLATE/config.yml
.github/ISSUE_TEMPLATE/feature_request.yml
.github/workflows/ci.yml
.github/workflows/gpu.yml
.github/workflows/nightly.yml
.github/workflows/publish.yml
.idea/.gitignore
.idea/misc.xml
.idea/modules.xml
.idea/vcs.xml
.idea/inspectionProfiles/Project_Default.xml
.idea/inspectionProfiles/profiles_settings.xml
docs/BENCHMARKS.md
docs/SERVER_RUN.md
docs/benchmarks_extending.md
docs/citing.md
docs/conf.py
docs/index.md
docs/v0.13-oracle-methodology.md
docs/ve-memory-model.md
docs/_templates/autosummary/class.rst
docs/_templates/autosummary/function.rst
docs/_templates/autosummary/module.rst
docs/api/index.md
docs/development/contributing.md
docs/development/index.md
docs/development/release.md
docs/development/testing.md
docs/examples/index.md
docs/getting_started/index.md
docs/getting_started/installation.md
docs/getting_started/quickstart.md
docs/user_guide/autograd.md
docs/user_guide/benchmarks.md
docs/user_guide/concepts.md
docs/user_guide/devices.md
docs/user_guide/index.md
docs/user_guide/inference.md
docs/user_guide/interventions.md
docs/user_guide/learning.md
docs/user_guide/mechanisms.md
docs/user_guide/persistence.md
docs/user_guide/probability_sets.md
docs/user_guide/sampling.md
docs/user_guide/scm.md
examples/01_parameter_learning.ipynb
examples/02_inference.ipynb
examples/03_interventions.ipynb
examples/04_sampling.ipynb
examples/05_updating.ipynb
examples/06_mechanism_gallery.ipynb
examples/07_batching_devices_persistence.ipynb
examples/08_autograd.ipynb
examples/09_discrete_noise_and_probability_sets.ipynb
examples/10_confounding_and_noise_diagnostics.ipynb
examples/11_continuous_noise_and_counterfactuals.ipynb
examples/12_mdp_trajectories_factored_sets.ipynb
examples/13_discrete_counterfactuals.ipynb
examples/14_structure_uncertainty_hypothesis_sets.ipynb
examples/15_agreement_rule.ipynb
examples/16_export_factored_sets.ipynb
examples/17_continuous_hypothesis_sets.ipynb
examples/18_confounded_rows.ipynb
examples/README.md
nbn/__init__.py
nbn/_mps_env.py
nbn/_version.py
nbn/config.py
nbn.egg-info/PKG-INFO
nbn.egg-info/SOURCES.txt
nbn.egg-info/dependency_links.txt
nbn.egg-info/entry_points.txt
nbn.egg-info/requires.txt
nbn.egg-info/scm_file_list.json
nbn.egg-info/scm_version.json
nbn.egg-info/top_level.txt
nbn/bench/__init__.py
nbn/bench/_env.py
nbn/bench/_extras.py
nbn/bench/_merge.py
nbn/bench/_paper_agg.py
nbn/bench/_paper_figures.py
nbn/bench/_paper_panels.py
nbn/bench/_run_logging.py
nbn/bench/calibration_diagnostic.py
nbn/bench/cli.py
nbn/bench/metrics.py
nbn/bench/synthetic.py
nbn/bench/adapters/__init__.py
nbn/bench/adapters/nbn_adapter.py
nbn/bench/adapters/pgmpy_adapter.py
nbn/bench/adapters/pomegranate_adapter.py
nbn/bench/adapters/pyro_adapter.py
nbn/bench/configs/bnlearn/complete/inference_complete.yaml
nbn/bench/configs/bnlearn/reruns/bnlearn_continuous.yaml
nbn/bench/configs/bnlearn/reruns/bnlearn_discrete_avi.yaml
nbn/bench/configs/bnlearn/smoke_tests/bug4_torture_smoke.yaml
nbn/bench/configs/bnlearn/smoke_tests/cardinality_smoke.yaml
nbn/bench/configs/bnlearn/smoke_tests/inference_smoke.yaml
nbn/bench/configs/synthetic/complete/inference_complete.yaml
nbn/bench/configs/synthetic/complete/inference_scalability_complete.yaml
nbn/bench/configs/synthetic/complete/parameter_learning_complete.yaml
nbn/bench/configs/synthetic/learning_curves/learning_curves.yaml
nbn/bench/configs/synthetic/learning_curves/learning_curves_smoke.yaml
nbn/bench/configs/synthetic/reruns/batch_speed_avi.yaml
nbn/bench/configs/synthetic/reruns/batch_speed_nbn.yaml
nbn/bench/configs/synthetic/reruns/complete_continuous.yaml
nbn/bench/configs/synthetic/reruns/complete_discrete_avi.yaml
nbn/bench/configs/synthetic/reruns/learning_curves_pgmpy_lg.yaml
nbn/bench/configs/synthetic/reruns/param_learning_pgmpy_lg.yaml
nbn/bench/configs/synthetic/reruns/scalability_continuous.yaml
nbn/bench/configs/synthetic/reruns/scalability_discrete_avi.yaml
nbn/bench/configs/synthetic/smoke_tests/inference_smoke.yaml
nbn/bench/configs/synthetic/smoke_tests/inference_smoke_random_only.yaml
nbn/bench/configs/synthetic/smoke_tests/inference_smoke_topological.yaml
nbn/bench/configs/synthetic/smoke_tests/parameter_learning_smoke.yaml
nbn/bench/configs/synthetic/smoke_tests/scalability_smoke.yaml
nbn/bench/configs/synthetic/speed/inference_speed.yaml
nbn/bench/configs/synthetic/speed/inference_speed_smoke.yaml
nbn/bench/configs/synthetic/speed/inference_speed_smoke2.yaml
nbn/bench/core/__init__.py
nbn/bench/core/_device.py
nbn/bench/core/applicability.py
nbn/bench/core/cell_runner.py
nbn/bench/core/cell_worker.py
nbn/bench/core/config.py
nbn/bench/core/cpt_extraction.py
nbn/bench/core/exact_discrete.py
nbn/bench/core/interfaces.py
nbn/bench/core/oracle.py
nbn/bench/core/output.py
nbn/bench/core/predictive_sampling.py
nbn/bench/core/results.py
nbn/bench/core/runner.py
nbn/bench/core/seeding.py
nbn/bench/core/yaml_config.py
nbn/bench/data/bnlearn/arth150.json
nbn/bench/data/bnlearn/ecoli70.json
nbn/bench/data/bnlearn/healthcare.json
nbn/bench/data/bnlearn/magic-irri.json
nbn/bench/data/bnlearn/magic-niab.json
nbn/bench/data/bnlearn/mehra.json
nbn/bench/data/bnlearn/sangiovese.json
nbn/bench/diagnostics/__init__.py
nbn/bench/diagnostics/triage_oom_rows.py
nbn/bench/diagnostics/ve_profile_n20.py
nbn/bench/domains/__init__.py
nbn/bench/domains/_n_parameters.py
nbn/bench/domains/base.py
nbn/bench/domains/posterior.py
nbn/bench/measurements/__init__.py
nbn/bench/measurements/_batch_dispatch.py
nbn/bench/measurements/accuracy_timing.py
nbn/bench/measurements/param_learning.py
nbn/bench/measurements/timing_only.py
nbn/bench/problems/__init__.py
nbn/bench/problems/bnlearn.py
nbn/bench/problems/synthetic.py
nbn/bench/selectors/__init__.py
nbn/bench/selectors/_batching.py
nbn/bench/selectors/heaviest.py
nbn/bench/selectors/topological.py
nbn/bench/selectors/uniform.py
nbn/core/__init__.py
nbn/core/dag.py
nbn/core/factor.py
nbn/core/network.py
nbn/core/query.py
nbn/core/variables.py
nbn/data/__init__.py
nbn/inference/__init__.py
nbn/inference/_cache.py
nbn/inference/_elimination_order.py
nbn/inference/_memory.py
nbn/inference/_pruning.py
nbn/inference/_validation.py
nbn/inference/_ve_memory.py
nbn/inference/amortized_is.py
nbn/inference/amortized_vi.py
nbn/inference/base.py
nbn/inference/hybrid.py
nbn/inference/likelihood_weighting.py
nbn/inference/recognition_net.py
nbn/inference/state.py
nbn/inference/tensor_ve.py
nbn/learning/__init__.py
nbn/learning/fit.py
nbn/learning/warm_start.py
nbn/learning/weighting.py
nbn/mechanisms/__init__.py
nbn/mechanisms/base.py
nbn/mechanisms/non_parametric/__init__.py
nbn/mechanisms/non_parametric/conditional_kde.py
nbn/mechanisms/non_parametric/flexcode.py
nbn/mechanisms/non_parametric/knn_conditional.py
nbn/mechanisms/non_parametric/smoothed_empirical_categorical.py
nbn/mechanisms/parametric/__init__.py
nbn/mechanisms/parametric/binning_categorical.py
nbn/mechanisms/parametric/categorical_table.py
nbn/mechanisms/parametric/deterministic.py
nbn/mechanisms/parametric/dirac_gaussian.py
nbn/mechanisms/parametric/linear_gaussian.py
nbn/mechanisms/parametric/mdn.py
nbn/mechanisms/parametric/neural_categorical.py
nbn/mechanisms/parametric/normalizing_flow.py
nbn/sampling/__init__.py
nbn/sampling/ancestral.py
nbn/scm/__init__.py
nbn/scm/_stats.py
nbn/scm/calibration.py
nbn/scm/categorical.py
nbn/scm/diagnostics.py
nbn/scm/episodes.py
nbn/scm/fitting.py
nbn/scm/flow.py
nbn/scm/mechanisms.py
nbn/scm/model.py
nbn/scm/noise.py
nbn/scm/sets.py
nbn/structure/__init__.py
nbn/structure/agreement.py
nbn/structure/batched.py
nbn/structure/continuous.py
nbn/structure/export.py
nbn/structure/hypotheses.py
nbn/structure/schemas/factored_sets_v1.schema.json
nbn/uncertainty/__init__.py
nbn/uncertainty/binomial.py
nbn/uncertainty/builders.py
nbn/uncertainty/confounded.py
nbn/uncertainty/counts.py
nbn/uncertainty/sets.py
nbn/update/__init__.py
nbn/update/base.py
nbn/update/dirichlet.py
nbn/update/online_laplace.py
nbn/update/orchestrate.py
nbn/update/recursive_gaussian.py
nbn/utils/__init__.py
nbn/utils/batching.py
nbn/utils/device.py
nbn/utils/einsum.py
nbn/utils/grad.py
nbn/utils/seed.py
nbn/utils/topo.py
scripts/bench_batched_candidates.py
scripts/bench_hypothesis_set.py
scripts/convert_bnlearn_continuous.R
scripts/run_all_benchmarks.sh
scripts/ve_memory_validation.py
tests/__init__.py
tests/conftest.py
tests/bench/__init__.py
tests/bench/_failing_cell_worker.py
tests/bench/conftest.py
tests/bench/test_adapter_contract.py
tests/bench/test_adapter_protocol_query_batch.py
tests/bench/test_adapters_phase3.py
tests/bench/test_applicability.py
tests/bench/test_bnlearn_cache_atomic.py
tests/bench/test_bnlearn_problem_source.py
tests/bench/test_cat_bayes_baseline.py
tests/bench/test_cell_memory_guard.py
tests/bench/test_cell_result_schema.py
tests/bench/test_cell_runner.py
tests/bench/test_cell_seeding.py
tests/bench/test_cli_pipeline.py
tests/bench/test_config_names_unique.py
tests/bench/test_device_autodetect.py
tests/bench/test_dirichlet_alpha_config.py
tests/bench/test_env_check.py
tests/bench/test_fit_once_save_reload.py
tests/bench/test_hashseed_independence.py
tests/bench/test_measurements.py
tests/bench/test_metrics_calibration.py
tests/bench/test_metrics_continuous_w1_ks.py
tests/bench/test_metrics_device_safety.py
tests/bench/test_metrics_identities.py
tests/bench/test_metrics_log_likelihood.py
tests/bench/test_metrics_param_recovery.py
tests/bench/test_metrics_per_node.py
tests/bench/test_n_parameters.py
tests/bench/test_n_train_sweep.py
tests/bench/test_nbn_adapter.py
tests/bench/test_nbn_calibration.py
tests/bench/test_nbn_param_recovery.py
tests/bench/test_nbn_score_data.py
tests/bench/test_oracle.py
tests/bench/test_oracle_exact_discrete.py
tests/bench/test_output.py
tests/bench/test_paper_agg.py
tests/bench/test_paper_figures.py
tests/bench/test_paper_figures_batch_speed.py
tests/bench/test_paper_panels.py
tests/bench/test_param_learning_calibration.py
tests/bench/test_param_learning_recovery.py
tests/bench/test_pgmpy_adapter.py
tests/bench/test_pgmpy_lg_sem.py
tests/bench/test_pgmpy_param_recovery.py
tests/bench/test_phase3_data_model.py
tests/bench/test_pomegranate_adapter.py
tests/bench/test_pomegranate_param_recovery.py
tests/bench/test_protocol_conformance.py
tests/bench/test_pyro_adapter.py
tests/bench/test_pyro_param_recovery.py
tests/bench/test_query_batch_contract.py
tests/bench/test_rerun_configs_and_merge.py
tests/bench/test_runner.py
tests/bench/test_runner_phase2_wiring.py
tests/bench/test_runner_robustness.py
tests/bench/test_runner_seed_skip.py
tests/bench/test_scoring_status_semantics.py
tests/bench/test_selector_heaviest.py
tests/bench/test_selector_n_batch_queries.py
tests/bench/test_selector_topological.py
tests/bench/test_synthetic_column_order.py
tests/bench/test_synthetic_generator.py
tests/bench/test_synthetic_source.py
tests/bench/test_triage_oom_rows.py
tests/bench/test_uniform_selector.py
tests/bench/test_yaml_config.py
tests/bench/integration/__init__.py
tests/bench/integration/test_bnlearn_sanity.py
tests/bench/integration/test_fit_once_sweep.py
tests/bench/integration/test_no_insertion_order_indexing.py
tests/bench/integration/test_runner_batch_dispatch.py
tests/bench/integration/test_speed_benchmark_smoke.py
tests/data/make_master_mdn_flow_checkpoint.py
tests/data/make_master_node_prefix_pickle.py
tests/data/master_mdn_flow_model.pt
tests/data/master_mdn_flow_ref.pt
tests/data/master_node_prefix_pickle.pt
tests/data/bnlearn/alarm.bif
tests/data/bnlearn/asia.bif
tests/data/bnlearn/sachs.bif
tests/integration/__init__.py
tests/integration/test_inference_neural_categorical.py
tests/integration/test_intervention_do_continuous.py
tests/perf/__init__.py
tests/perf/test_inference_speed.py
tests/perf/test_selector_speed.py
tests/perf/test_ve_memory_validation.py
tests/perf/test_ve_pruning_speed.py
tests/unit/__init__.py
tests/unit/test_ais_per_query_fallback.py
tests/unit/test_amortized_is_engine.py
tests/unit/test_amortized_vi_engine.py
tests/unit/test_ancestral_sampler_parent_dependence.py
tests/unit/test_avi_quality_gate.py
tests/unit/test_binning_categorical_no_contamination.py
tests/unit/test_categorical_cardinality.py
tests/unit/test_dag.py
tests/unit/test_device_dtype_casts.py
tests/unit/test_device_propagation.py
tests/unit/test_dirac_gaussian.py
tests/unit/test_do_operator.py
tests/unit/test_dotted_node_names.py
tests/unit/test_fisher_vmap.py
tests/unit/test_fit_training_budgets.py
tests/unit/test_flow_early_stopping.py
tests/unit/test_from_bif.py
tests/unit/test_hybrid_router.py
tests/unit/test_issues_304_305.py
tests/unit/test_lw_query_inference_mode.py
tests/unit/test_mdn_robustness.py
tests/unit/test_mechanism_conformance.py
tests/unit/test_mechanisms.py
tests/unit/test_min_fill_order.py
tests/unit/test_model_lifecycle_contracts.py
tests/unit/test_model_log_prob.py
tests/unit/test_mps_compat.py
tests/unit/test_nonparametric_mechanisms.py
tests/unit/test_nonparametric_query_cost.py
tests/unit/test_parameter_snapshot_contract.py
tests/unit/test_parent_conditioned_proposal.py
tests/unit/test_parent_gradient_contract.py
tests/unit/test_pruning.py
tests/unit/test_psis_khat.py
tests/unit/test_query_ess.py
tests/unit/test_resolve_device.py
tests/unit/test_review_fixes_core.py
tests/unit/test_review_fixes_inference.py
tests/unit/test_review_fixes_inference_r2.py
tests/unit/test_review_fixes_inference_r3.py
tests/unit/test_review_fixes_inference_r4.py
tests/unit/test_review_fixes_mechanisms.py
tests/unit/test_update_exact.py
tests/unit/test_update_neural_ewc.py
tests/unit/test_update_nonparametric.py
tests/unit/test_ve_batch_chunking.py
tests/unit/test_ve_memory_guard.py
tests/unit/test_ve_memory_model.py
tests/unit/test_vectorized_query_batch_correctness.py
tests/unit/test_warm_start_contract.py
tests/unit/test_weighted_fitting.py
tests/unit/scm/__init__.py
tests/unit/scm/confounded_sim.py
tests/unit/scm/simulators.py
tests/unit/scm/test_binomial.py
tests/unit/scm/test_categorical.py
tests/unit/scm/test_confounded.py
tests/unit/scm/test_diagnostics.py
tests/unit/scm/test_mechanisms.py
tests/unit/scm/test_model.py
tests/unit/scm/test_neural.py
tests/unit/scm/test_probability_set_coverage.py
tests/unit/scm/test_review_regressions.py
tests/unit/scm/test_sets.py
tests/unit/scm/test_sets_from_counts.py
tests/unit/structure/__init__.py
tests/unit/structure/test_agreement.py
tests/unit/structure/test_continuous.py
tests/unit/structure/test_export.py
tests/unit/structure/test_hypothesis_set.py