.gitignore
LICENSE
README.md
RESEARCH.md
mypy.ini
pyproject.toml
ruff.toml
.github/workflows/ci.yml
.github/workflows/nightly.yml
.github/workflows/publish.yml
.idea/.gitignore
.idea/NeuralBayesianNetworks.iml
.idea/misc.xml
.idea/modules.xml
.idea/vcs.xml
.idea/inspectionProfiles/Project_Default.xml
.idea/inspectionProfiles/profiles_settings.xml
docs/SERVER_RUN.md
docs/benchmarks_extending.md
docs/bnlearn-complete-launch-plan.md
docs/phase2-design-draft.md
docs/phase3-design-draft.md
docs/phase4-design-draft.md
docs/v0.13-benchmark-redesign.md
docs/v0.13-bug2-subnetwork-pruning.md
docs/v0.13-nbn-cat-ve-investigation.md
docs/v0.13-oracle-methodology.md
docs/v0.13-paper-figures.md
docs/v0.13-runner-subprocess-isolation.md
docs/v0.14-batched-inference-engines-research.md
docs/v0.14-batched-queries-design.md
docs/v0.14-design-decisions.md
docs/v0.14-fit-once-query-many-design.md
docs/v0.14-lw-batch-sampling-investigation.md
docs/audits/v0.12-bug-b-and-c-mdn-overflow.md
docs/audits/v0.12-pyro-gpu-investigation.md
docs/audits/v0.7-43-fit-path-audit.md
docs/audits/v0.8-baseline-applicability-audit.md
docs/v0.6b/profiler_trace.json
docs/v0.6b/profiler_trace.md
docs/v0.6c-a/findings.md
docs/v0.6c-a/lw_b_scaling.json
docs/v0.6c-a/mdn_simplex.json
docs/v0.6c-d/dnf_cells.md
docs/v0.6c-d/run_summary.md
nbn/__init__.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/_aggregate.py
nbn/bench/_extras.py
nbn/bench/_paper_figures.py
nbn/bench/_plot_v2.py
nbn/bench/_run_logging.py
nbn/bench/_tables.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/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/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/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/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/_elimination_order.py
nbn/inference/_pruning.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/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/seed.py
nbn/utils/topo.py
notebooks/README.md
notebooks/_make_notebooks.py
notebooks/bnlearn_inference_complete.ipynb
notebooks/inference_complete.ipynb
notebooks/inference_scalability_complete.ipynb
notebooks/inference_speed.ipynb
notebooks/learning_curves.ipynb
notebooks/parameter_learning_complete.ipynb
scripts/convert_bnlearn_continuous.R
scripts/make_paper_figures.py
scripts/run_all_benchmarks.sh
tests/__init__.py
tests/conftest.py
tests/integration/__init__.py
tests/integration/test_backwards_compat_synthetic_shim.py
tests/integration/test_baseline_applicability_consolidation.py
tests/integration/test_bnlearn_sanity.py
tests/integration/test_fit_once_sweep.py
tests/integration/test_inference_neural_categorical.py
tests/integration/test_intervention_do_continuous.py
tests/integration/test_no_insertion_order_indexing.py
tests/integration/test_runner_batch_dispatch.py
tests/integration/test_speed_benchmark_smoke.py
tests/perf/__init__.py
tests/unit/__init__.py
tests/unit/test_adapter_protocol_query_batch.py
tests/unit/test_aggregator.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_binning_categorical_no_contamination.py
tests/unit/test_cat_bayes_baseline.py
tests/unit/test_dag.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_fit_training_budgets.py
tests/unit/test_from_bif.py
tests/unit/test_hybrid_router.py
tests/unit/test_lw_query_inference_mode.py
tests/unit/test_mdn_robustness.py
tests/unit/test_mechanisms.py
tests/unit/test_metrics.py
tests/unit/test_metrics_calibration.py
tests/unit/test_metrics_continuous.py
tests/unit/test_metrics_log_likelihood.py
tests/unit/test_metrics_param_recovery.py
tests/unit/test_metrics_per_node.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_nbn_query_batch.py
tests/unit/test_nonparametric_mechanisms.py
tests/unit/test_paper_figures_batch_speed.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_plotter_legend_filtering.py
tests/unit/test_pomegranate_query_batch.py
tests/unit/test_psis_khat.py
tests/unit/test_query_ess.py
tests/unit/test_selector_n_batch_queries.py
tests/unit/test_synthetic_column_order.py
tests/unit/test_synthetic_generator.py
tests/unit/test_tables.py
tests/unit/test_triage_oom_rows.py
tests/unit/test_update_exact.py
tests/unit/test_update_neural_ewc.py
tests/unit/test_ve_batch_chunking.py
tests/unit/test_ve_memory_guard.py
tests/unit/test_vectorized_query_batch_correctness.py
tests/unit/test_warm_start_contract.py
tests/unit/test_weighted_fitting.py
tests/unit/v013/__init__.py
tests/unit/v013/_failing_cell_worker.py
tests/unit/v013/conftest.py
tests/unit/v013/test_adapters_phase3.py
tests/unit/v013/test_aggregator.py
tests/unit/v013/test_bnlearn_problem_source.py
tests/unit/v013/test_categorical_cardinality.py
tests/unit/v013/test_cell_memory_guard.py
tests/unit/v013/test_cell_result_schema.py
tests/unit/v013/test_cell_runner.py
tests/unit/v013/test_cli_pipeline.py
tests/unit/v013/test_device_autodetect.py
tests/unit/v013/test_fit_once_save_reload.py
tests/unit/v013/test_gpytorch_adapter.py
tests/unit/v013/test_interfaces_structural.py
tests/unit/v013/test_measurements.py
tests/unit/v013/test_metrics_device_safety.py
tests/unit/v013/test_n_parameters.py
tests/unit/v013/test_n_train_sweep.py
tests/unit/v013/test_nbn_adapter.py
tests/unit/v013/test_nbn_calibration.py
tests/unit/v013/test_nbn_param_recovery.py
tests/unit/v013/test_nbn_score_data.py
tests/unit/v013/test_oracle.py
tests/unit/v013/test_output.py
tests/unit/v013/test_paper_figures.py
tests/unit/v013/test_param_learning_calibration.py
tests/unit/v013/test_param_learning_recovery.py
tests/unit/v013/test_pgmpy_adapter.py
tests/unit/v013/test_pgmpy_lg_sem.py
tests/unit/v013/test_pgmpy_param_recovery.py
tests/unit/v013/test_phase3_data_model.py
tests/unit/v013/test_pomegranate_adapter.py
tests/unit/v013/test_pomegranate_param_recovery.py
tests/unit/v013/test_pruning.py
tests/unit/v013/test_pyro_adapter.py
tests/unit/v013/test_pyro_param_recovery.py
tests/unit/v013/test_runner.py
tests/unit/v013/test_runner_phase2_wiring.py
tests/unit/v013/test_runner_robustness.py
tests/unit/v013/test_runner_seed_skip.py
tests/unit/v013/test_selector_heaviest.py
tests/unit/v013/test_selector_topological.py
tests/unit/v013/test_synthetic_source.py
tests/unit/v013/test_uniform_selector.py
tests/unit/v013/test_yaml_config.py