LICENSE
README.md
pyproject.toml
bayesflow/__init__.py
bayesflow.egg-info/PKG-INFO
bayesflow.egg-info/SOURCES.txt
bayesflow.egg-info/dependency_links.txt
bayesflow.egg-info/requires.txt
bayesflow.egg-info/top_level.txt
bayesflow/adapters/__init__.py
bayesflow/adapters/adapter.py
bayesflow/adapters/transforms/__init__.py
bayesflow/adapters/transforms/as_set.py
bayesflow/adapters/transforms/as_time_series.py
bayesflow/adapters/transforms/broadcast.py
bayesflow/adapters/transforms/concatenate.py
bayesflow/adapters/transforms/constrain.py
bayesflow/adapters/transforms/convert_dtype.py
bayesflow/adapters/transforms/drop.py
bayesflow/adapters/transforms/elementwise_transform.py
bayesflow/adapters/transforms/expand_dims.py
bayesflow/adapters/transforms/filter_transform.py
bayesflow/adapters/transforms/group.py
bayesflow/adapters/transforms/keep.py
bayesflow/adapters/transforms/log.py
bayesflow/adapters/transforms/map_transform.py
bayesflow/adapters/transforms/nan_to_num.py
bayesflow/adapters/transforms/numpy_transform.py
bayesflow/adapters/transforms/one_hot.py
bayesflow/adapters/transforms/random_subsample.py
bayesflow/adapters/transforms/rename.py
bayesflow/adapters/transforms/scale.py
bayesflow/adapters/transforms/serializable_custom_transform.py
bayesflow/adapters/transforms/shift.py
bayesflow/adapters/transforms/split.py
bayesflow/adapters/transforms/sqrt.py
bayesflow/adapters/transforms/squeeze.py
bayesflow/adapters/transforms/standardize.py
bayesflow/adapters/transforms/take.py
bayesflow/adapters/transforms/to_array.py
bayesflow/adapters/transforms/to_dict.py
bayesflow/adapters/transforms/transform.py
bayesflow/adapters/transforms/ungroup.py
bayesflow/approximators/__init__.py
bayesflow/approximators/approximator.py
bayesflow/approximators/compositional_approximator.py
bayesflow/approximators/continuous_approximator.py
bayesflow/approximators/ensemble_approximator.py
bayesflow/approximators/model_comparison_approximator.py
bayesflow/approximators/ratio_approximator.py
bayesflow/approximators/scoring_rule_approximator.py
bayesflow/approximators/backend_approximators/__init__.py
bayesflow/approximators/backend_approximators/backend_approximator.py
bayesflow/approximators/backend_approximators/jax_approximator.py
bayesflow/approximators/backend_approximators/tensorflow_approximator.py
bayesflow/approximators/backend_approximators/torch_approximator.py
bayesflow/approximators/helpers/__init__.py
bayesflow/approximators/helpers/compositional.py
bayesflow/approximators/helpers/conditions.py
bayesflow/approximators/helpers/samplers.py
bayesflow/augmentations/__init__.py
bayesflow/augmentations/nnpe.py
bayesflow/datasets/__init__.py
bayesflow/datasets/disk_dataset.py
bayesflow/datasets/ensemble_dataset.py
bayesflow/datasets/ensemble_indexed_dataset.py
bayesflow/datasets/ensemble_online_dataset.py
bayesflow/datasets/offline_dataset.py
bayesflow/datasets/online_dataset.py
bayesflow/datasets/helpers/__init__.py
bayesflow/datasets/helpers/augmentations.py
bayesflow/datasets/helpers/ensemble_sharing.py
bayesflow/diagnostics/__init__.py
bayesflow/diagnostics/metrics/__init__.py
bayesflow/diagnostics/metrics/calibration_error.py
bayesflow/diagnostics/metrics/calibration_log_gamma.py
bayesflow/diagnostics/metrics/classifier_two_sample_test.py
bayesflow/diagnostics/metrics/correlation.py
bayesflow/diagnostics/metrics/expected_calibration_error.py
bayesflow/diagnostics/metrics/model_misspecification.py
bayesflow/diagnostics/metrics/posterior_contraction.py
bayesflow/diagnostics/metrics/posterior_z_score.py
bayesflow/diagnostics/metrics/root_mean_squared_error.py
bayesflow/diagnostics/plots/__init__.py
bayesflow/diagnostics/plots/calibration_ecdf.py
bayesflow/diagnostics/plots/calibration_ecdf_from_quantiles.py
bayesflow/diagnostics/plots/calibration_histogram.py
bayesflow/diagnostics/plots/coverage.py
bayesflow/diagnostics/plots/loss.py
bayesflow/diagnostics/plots/mc_calibration.py
bayesflow/diagnostics/plots/mc_confusion_matrix.py
bayesflow/diagnostics/plots/mmd_hypothesis_test.py
bayesflow/diagnostics/plots/pairs_posterior.py
bayesflow/diagnostics/plots/pairs_quantity.py
bayesflow/diagnostics/plots/pairs_samples.py
bayesflow/diagnostics/plots/plot_quantity.py
bayesflow/diagnostics/plots/recovery.py
bayesflow/diagnostics/plots/recovery_from_estimates.py
bayesflow/diagnostics/plots/z_score_contraction.py
bayesflow/distributions/__init__.py
bayesflow/distributions/diagonal_normal.py
bayesflow/distributions/diagonal_student_t.py
bayesflow/distributions/distribution.py
bayesflow/distributions/mixture.py
bayesflow/experimental/__init__.py
bayesflow/experimental/free_form_flow/__init__.py
bayesflow/experimental/free_form_flow/free_form_flow.py
bayesflow/experimental/graphical_approximator/__init__.py
bayesflow/experimental/graphical_approximator/graphical_approximator.py
bayesflow/experimental/graphical_approximator/utils.py
bayesflow/experimental/graphical_approximator/example_approximators/__init__.py
bayesflow/experimental/graphical_approximator/example_approximators/crossed_design_irt_approximator.py
bayesflow/experimental/graphical_approximator/example_approximators/single_level_approximator.py
bayesflow/experimental/graphical_approximator/example_approximators/three_level_approximator.py
bayesflow/experimental/graphical_approximator/example_approximators/two_level_approximator.py
bayesflow/experimental/graphical_simulator/__init__.py
bayesflow/experimental/graphical_simulator/graphical_simulator.py
bayesflow/experimental/graphical_simulator/example_simulators/__init__.py
bayesflow/experimental/graphical_simulator/example_simulators/crossed_design_irt_simulator.py
bayesflow/experimental/graphical_simulator/example_simulators/single_level_simulator.py
bayesflow/experimental/graphical_simulator/example_simulators/three_level_simulator.py
bayesflow/experimental/graphical_simulator/example_simulators/two_level_simulator.py
bayesflow/experimental/graphs/__init__.py
bayesflow/experimental/graphs/expanded_graph.py
bayesflow/experimental/graphs/inverted_graph.py
bayesflow/experimental/graphs/simulation_graph.py
bayesflow/experimental/graphs/utils.py
bayesflow/links/__init__.py
bayesflow/links/cholesky_factor.py
bayesflow/links/ordered.py
bayesflow/links/ordered_quantiles.py
bayesflow/links/positive_definite.py
bayesflow/metrics/__init__.py
bayesflow/metrics/maximum_mean_discrepancy.py
bayesflow/metrics/root_mean_squard_error.py
bayesflow/metrics/functional/__init__.py
bayesflow/metrics/functional/kernels.py
bayesflow/metrics/functional/maximum_mean_discrepancy.py
bayesflow/metrics/functional/root_mean_squared_error.py
bayesflow/networks/__init__.py
bayesflow/networks/defaults.py
bayesflow/networks/helpers/__init__.py
bayesflow/networks/helpers/blocks/__init__.py
bayesflow/networks/helpers/blocks/conditional_dense_block.py
bayesflow/networks/helpers/blocks/dense_block.py
bayesflow/networks/helpers/embeddings/__init__.py
bayesflow/networks/helpers/embeddings/featurewise_linear_modulation.py
bayesflow/networks/helpers/embeddings/fourier_embedding.py
bayesflow/networks/helpers/embeddings/recurrent_embedding.py
bayesflow/networks/helpers/embeddings/time2vec.py
bayesflow/networks/helpers/norms/__init__.py
bayesflow/networks/helpers/norms/simple_norm.py
bayesflow/networks/helpers/sequential/__init__.py
bayesflow/networks/helpers/sequential/sequential.py
bayesflow/networks/helpers/standardization/__init__.py
bayesflow/networks/helpers/standardization/standardization.py
bayesflow/networks/helpers/standardization/standardize.py
bayesflow/networks/inference/__init__.py
bayesflow/networks/inference/inference_network.py
bayesflow/networks/inference/consistency/__init__.py
bayesflow/networks/inference/consistency/consistency_model.py
bayesflow/networks/inference/consistency/stable_consistency_model.py
bayesflow/networks/inference/coupling/__init__.py
bayesflow/networks/inference/coupling/actnorm.py
bayesflow/networks/inference/coupling/coupling_flow.py
bayesflow/networks/inference/coupling/invertible_layer.py
bayesflow/networks/inference/coupling/layers/__init__.py
bayesflow/networks/inference/coupling/layers/dual_coupling.py
bayesflow/networks/inference/coupling/layers/single_coupling.py
bayesflow/networks/inference/coupling/permutations/__init__.py
bayesflow/networks/inference/coupling/permutations/fixed_permutation.py
bayesflow/networks/inference/coupling/permutations/orthogonal.py
bayesflow/networks/inference/coupling/permutations/random.py
bayesflow/networks/inference/coupling/permutations/swap.py
bayesflow/networks/inference/coupling/transforms/__init__.py
bayesflow/networks/inference/coupling/transforms/_rational_quadratic.py
bayesflow/networks/inference/coupling/transforms/affine_transform.py
bayesflow/networks/inference/coupling/transforms/spline_transform.py
bayesflow/networks/inference/coupling/transforms/transform.py
bayesflow/networks/inference/diffusion/__init__.py
bayesflow/networks/inference/diffusion/diffusion_model.py
bayesflow/networks/inference/diffusion/dispatch.py
bayesflow/networks/inference/diffusion/schedules/__init__.py
bayesflow/networks/inference/diffusion/schedules/cosine_noise_schedule.py
bayesflow/networks/inference/diffusion/schedules/edm_noise_schedule.py
bayesflow/networks/inference/diffusion/schedules/noise_schedule.py
bayesflow/networks/inference/flow_matching/__init__.py
bayesflow/networks/inference/flow_matching/flow_matching.py
bayesflow/networks/inference/scoring/__init__.py
bayesflow/networks/inference/scoring/point_network.py
bayesflow/networks/inference/scoring/scoring_rule_network.py
bayesflow/networks/inference/scoring/scoring_rules/__init__.py
bayesflow/networks/inference/scoring/scoring_rules/cross_entropy_score.py
bayesflow/networks/inference/scoring/scoring_rules/mean_score.py
bayesflow/networks/inference/scoring/scoring_rules/median_score.py
bayesflow/networks/inference/scoring/scoring_rules/mixture_score.py
bayesflow/networks/inference/scoring/scoring_rules/mv_normal_score.py
bayesflow/networks/inference/scoring/scoring_rules/normed_difference_score.py
bayesflow/networks/inference/scoring/scoring_rules/parametric_distribution_score.py
bayesflow/networks/inference/scoring/scoring_rules/quantile_score.py
bayesflow/networks/inference/scoring/scoring_rules/scoring_rule.py
bayesflow/networks/subnets/__init__.py
bayesflow/networks/subnets/mlp/__init__.py
bayesflow/networks/subnets/mlp/mlp.py
bayesflow/networks/subnets/mlp/time_mlp.py
bayesflow/networks/subnets/unet/__init__.py
bayesflow/networks/subnets/unet/resuvit.py
bayesflow/networks/subnets/unet/unet.py
bayesflow/networks/subnets/unet/uvit.py
bayesflow/networks/subnets/unet/blocks/__init__.py
bayesflow/networks/subnets/unet/blocks/attention.py
bayesflow/networks/subnets/unet/blocks/downsample.py
bayesflow/networks/subnets/unet/blocks/residual.py
bayesflow/networks/subnets/unet/blocks/time_dense.py
bayesflow/networks/subnets/unet/blocks/transformer.py
bayesflow/networks/subnets/unet/blocks/upsample.py
bayesflow/networks/subnets/unet/embeddings/__init__.py
bayesflow/networks/subnets/unet/embeddings/dense_fourier.py
bayesflow/networks/summary/__init__.py
bayesflow/networks/summary/summary_network.py
bayesflow/networks/summary/convolutional/__init__.py
bayesflow/networks/summary/convolutional/convolutional_network.py
bayesflow/networks/summary/convolutional/double_conv.py
bayesflow/networks/summary/deep_set/__init__.py
bayesflow/networks/summary/deep_set/deep_set.py
bayesflow/networks/summary/deep_set/equivariant_layer.py
bayesflow/networks/summary/deep_set/invariant_layer.py
bayesflow/networks/summary/fusion/__init__.py
bayesflow/networks/summary/fusion/fusion_network.py
bayesflow/networks/summary/recurrent/__init__.py
bayesflow/networks/summary/recurrent/skip_recurrent.py
bayesflow/networks/summary/recurrent/time_series_network.py
bayesflow/networks/summary/transformers/__init__.py
bayesflow/networks/summary/transformers/fusion_transformer.py
bayesflow/networks/summary/transformers/set_transformer.py
bayesflow/networks/summary/transformers/time_series_transformer.py
bayesflow/networks/summary/transformers/transformer.py
bayesflow/networks/summary/transformers/attention/__init__.py
bayesflow/networks/summary/transformers/attention/feedforward_net.py
bayesflow/networks/summary/transformers/attention/induced_set_attention.py
bayesflow/networks/summary/transformers/attention/multihead_attention.py
bayesflow/networks/summary/transformers/attention/pooling_by_multihead_attention.py
bayesflow/networks/summary/transformers/attention/set_attention.py
bayesflow/scoring_rules/__init__.py
bayesflow/simulators/__init__.py
bayesflow/simulators/hierarchical_simulator.py
bayesflow/simulators/lambda_simulator.py
bayesflow/simulators/make_simulator.py
bayesflow/simulators/model_comparison_simulator.py
bayesflow/simulators/sequential_simulator.py
bayesflow/simulators/simulator.py
bayesflow/simulators/benchmark_simulators/__init__.py
bayesflow/simulators/benchmark_simulators/benchmark_simulator.py
bayesflow/simulators/benchmark_simulators/bernoulli_glm.py
bayesflow/simulators/benchmark_simulators/bernoulli_glm_raw.py
bayesflow/simulators/benchmark_simulators/gaussian_linear.py
bayesflow/simulators/benchmark_simulators/gaussian_linear_uniform.py
bayesflow/simulators/benchmark_simulators/gaussian_mixture.py
bayesflow/simulators/benchmark_simulators/inverse_kinematics.py
bayesflow/simulators/benchmark_simulators/lotka_volterra.py
bayesflow/simulators/benchmark_simulators/sir.py
bayesflow/simulators/benchmark_simulators/slcp.py
bayesflow/simulators/benchmark_simulators/slcp_distractors.py
bayesflow/simulators/benchmark_simulators/two_moons.py
bayesflow/types/__init__.py
bayesflow/types/shape.py
bayesflow/types/tensor.py
bayesflow/utils/__init__.py
bayesflow/utils/callbacks.py
bayesflow/utils/context_managers.py
bayesflow/utils/decorators.py
bayesflow/utils/devices.py
bayesflow/utils/dict_utils.py
bayesflow/utils/empty.py
bayesflow/utils/functional.py
bayesflow/utils/git.py
bayesflow/utils/hparam_utils.py
bayesflow/utils/integrate.py
bayesflow/utils/io.py
bayesflow/utils/keras_utils.py
bayesflow/utils/logging.py
bayesflow/utils/numpy_utils.py
bayesflow/utils/plot_utils.py
bayesflow/utils/serialization.py
bayesflow/utils/tensor_utils.py
bayesflow/utils/tree.py
bayesflow/utils/validators.py
bayesflow/utils/_docs/__init__.py
bayesflow/utils/_docs/_populate_all.py
bayesflow/utils/classification/__init__.py
bayesflow/utils/classification/calibration_curve.py
bayesflow/utils/classification/confusion_matrix.py
bayesflow/utils/dispatch/__init__.py
bayesflow/utils/dispatch/find_cost.py
bayesflow/utils/dispatch/find_distribution.py
bayesflow/utils/dispatch/find_inference_network.py
bayesflow/utils/dispatch/find_network.py
bayesflow/utils/dispatch/find_permutation.py
bayesflow/utils/dispatch/find_pooling.py
bayesflow/utils/dispatch/find_recurrent_net.py
bayesflow/utils/dispatch/find_summary_network.py
bayesflow/utils/ecdf/__init__.py
bayesflow/utils/ecdf/minimal_coverage_probs.py
bayesflow/utils/ecdf/pointwise_ecdf_bands.py
bayesflow/utils/ecdf/ranks.py
bayesflow/utils/ecdf/simultaneous_ecdf_bands.py
bayesflow/utils/exceptions/__init__.py
bayesflow/utils/exceptions/shape_error.py
bayesflow/utils/jacobian/__init__.py
bayesflow/utils/jacobian/jacobian.py
bayesflow/utils/jacobian/jacobian_trace.py
bayesflow/utils/jacobian/jvp.py
bayesflow/utils/jacobian/vjp.py
bayesflow/utils/optimal_transport/__init__.py
bayesflow/utils/optimal_transport/log_sinkhorn.py
bayesflow/utils/optimal_transport/optimal_transport.py
bayesflow/utils/optimal_transport/ot_utils.py
bayesflow/utils/optimal_transport/sinkhorn.py
bayesflow/workflows/__init__.py
bayesflow/workflows/basic_workflow.py
bayesflow/workflows/compositional_workflow.py
bayesflow/workflows/ensemble_workflow.py
bayesflow/workflows/workflow.py
bayesflow/wrappers/__init__.py
bayesflow/wrappers/mamba/__init__.py
bayesflow/wrappers/mamba/mamba.py
bayesflow/wrappers/mamba/mamba_block.py
bayesflow/wrappers/pymc/__init__.py
bayesflow/wrappers/pymc/neural_distribution.py
bayesflow/wrappers/pymc/pytensor_ops.py
bayesflow/wrappers/pymc/ratio_distribution.py
bayesflow/wrappers/pymc/backend/__init__.py
bayesflow/wrappers/pymc/backend/backend_ratio.py
bayesflow/wrappers/pymc/backend/jax_continuous.py
bayesflow/wrappers/pymc/backend/jax_dispatch.py
bayesflow/wrappers/pymc/backend/jax_ratio.py
bayesflow/wrappers/pymc/backend/jax_wrapper.py